#pragma once #include "core/omp_config.h" #include "utils/vector.h" #include "utils/matrix.h" #include "modules/neural_networks/layers/Layer.h" namespace neural_networks { template struct Model { std::vector*> layers; Model() = default; void add(Layer& layer) { layers.push_back(&layer); } }; } // end namespace neural_networks