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

Fix missing activation in batch norm

Browse Source
This commit is contained in:
Matúš Námešný
2024-06-02 14:41:14 +02:00
parent 7f745027cd
commit a4a2801ee4
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/batch_norm.cu
View File

@@ -190,5 +190,7 @@ float *BatchNorm2d::forward(const float *d_input) {
CUDA_CHECK(cudaGetLastError());
}
activation->activate(d_output);
return d_output;
}