[batchnorm] Bug fix for batch norm layer

Bug fix for requested variable for batch norm layer.

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>

diff --git a/nntrainer/layers/bn_layer.cpp b/nntrainer/layers/bn_layer.cpp
index 91e10a0..3cbe8c6 100644 (file)
--- a/nntrainer/layers/bn_layer.cpp
+++ b/nntrainer/layers/bn_layer.cpp
@@ -65,7 +65,7 @@ void BatchNormalizationLayer::finalize(InitLayerContext &context) {
     context.getName() + ":moving_mean", false);
   wt_idx[BNParams::var] = context.requestWeight(
     dim, initializers[BNParams::var], WeightRegularizer::NONE, 1.0f,
-    "BN::moving_variance", false);
+    context.getName() + ":moving_variance", false);
   wt_idx[BNParams::gamma] = context.requestWeight(
     dim, initializers[BNParams::gamma], WeightRegularizer::NONE, 1.0f,
     context.getName() + ":gamma", true);