We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent fd77a82 commit 5d60704Copy full SHA for 5d60704
1 file changed
include/layers/BatchNormalizationLayer.hpp
@@ -12,7 +12,8 @@ class BatchNormalizationLayer : public Layer {
12
const Tensor& mean, const Tensor& var,
13
float epsilon = 1e-5F, float momentum = 0.9F,
14
bool training_mode = false)
15
- : Layer(kBatchNormalization), scale_(scale),
+ : Layer(kBatchNormalization),
16
+ scale_(scale),
17
bias_(bias),
18
mean_(mean),
19
var_(var),
0 commit comments