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

Rename batchnorm

Browse Source
This commit is contained in:
Matúš Námešný
2024-05-20 13:05:48 +02:00
parent 94df063dcd
commit 6f8b5f4081
4 changed files with 20 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
include/layers/batch_norm.cuh
View File

@@ -8,11 +8,11 @@
namespace CUDANet::Layers {
class BatchNorm : public WeightedLayer {
class BatchNorm2D : public WeightedLayer {
public:
BatchNorm(int inputSize, int inputChannels, float epsilon, ActivationType activationType);
BatchNorm2D(int inputSize, int inputChannels, float epsilon, ActivationType activationType);
~BatchNorm();
~BatchNorm2D();
/**
* @brief Compute the forward pass of the batchnorm layer
@@ -105,13 +105,13 @@ class BatchNorm : public WeightedLayer {
/**
* @brief Initialize mean of the batchnorm layer with zeros
*
*
*/
void initializeMean();
/**
* @brief Initialize sqrt of variance of the batchnorm layer with ones
*
*
*/
void initializeSqrtVar();