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

Fix model weights export

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-21 00:05:56 +02:00
parent 9c5d853b75
commit d64a28bc9c
3 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/model/model.cpp
View File

@@ -43,6 +43,7 @@ float* Model::predict(const float* input) {
float* d_input = inputLayer->forward(input);
for (auto& layer : layers) {
std::cout << layer << std::endl;
d_input = layer->forward(d_input);
}