Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 17:54:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add layer name to vector

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-21 12:20:02 +02:00
parent 0170afaf3f
commit 942ee6a32b
2 changed files with 13 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
include/model/model.hpp
View File

@@ -11,7 +11,10 @@
namespace CUDANet {
enum TensorType { WEIGHT, BIAS, };
enum TensorType {
WEIGHT,
BIAS,
};
struct TensorInfo {
std::string name;
@@ -42,8 +45,8 @@ class Model {
int outputSize;
std::vector<Layers::SequentialLayer*> layers;
std::unordered_map<std::string, Layers::SequentialLayer*> layerMap;
std::vector<std::pair<std::string, Layers::SequentialLayer*>> layers;
std::unordered_map<std::string, Layers::SequentialLayer*> layerMap;
};
} // namespace CUDANet