30 lines
424 B
C++
30 lines
424 B
C++
#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 <typename T>
|
|
struct Model {
|
|
|
|
std::vector<Layer<T>*> layers;
|
|
|
|
Model() = default;
|
|
|
|
void add(Layer<T>& layer) {
|
|
layers.push_back(&layer);
|
|
}
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
} // end namespace neural_networks
|