Mathis/CUDANet
1
0
Fork 0
mirror of https://github.com/lordmathis/CUDANet.git synced 2025-11-06 01:34:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix adding module layers

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-29 21:02:26 +02:00
parent 4159454842
commit 8ac2da004c
4 changed files with 31 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
include/model/module.hpp
View File

@@ -11,8 +11,6 @@ namespace CUDANet {
class Module : public Layers::SequentialLayer {
public:
virtual ~Module() = 0;
virtual float* forward(const float* d_input) = 0;
int getOutputSize();