mirror of
https://github.com/lordmathis/CUDANet.git
synced 2025-11-06 01:34:22 +00:00
34 lines
603 B
C++
34 lines
603 B
C++
#ifndef CUDANET_MODEL_H
|
|
#define CUDANET_MODEL_H
|
|
|
|
#include <string>
|
|
#include <vector>
|
|
#include <map>
|
|
#include "layer.cuh"
|
|
|
|
namespace CUDANet {
|
|
|
|
class Model {
|
|
public:
|
|
Model(const int inputSize, const int inputChannels);
|
|
~Model();
|
|
|
|
float* predict(const float* input);
|
|
|
|
void addLayer(const std::string& name, Layers::SequentialLayer* layer);
|
|
|
|
private:
|
|
|
|
int inputSize;
|
|
int inputChannels;
|
|
|
|
int outputSize;
|
|
|
|
std::vector<Layers::SequentialLayer*> layers;
|
|
std::map<std::string, Layers::WeightedLayer*> layerMap;
|
|
|
|
};
|
|
|
|
} // namespace CUDANet
|
|
|
|
#endif // CUDANET_MODEL_H
|