Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-07 02:04:26 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement getOutputSize and getInputSize for seq layers

Browse Source
This commit is contained in:
Matúš Námešný
2024-04-22 20:31:58 +02:00
parent fbf6c44bdd
commit f17debc244
18 changed files with 186 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/model/model.cpp
View File

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