Load running mean and var from weight file

include/layers/batch_norm.cuh

@@ -57,6 +57,12 @@ class BatchNorm2d : public WeightedLayer, public TwoDLayer {
      */
     void setRunningMean(const float* running_mean_input);
 
+    /**
+     * @brief Get the Running Mean
+     * 
+     */
+    std::vector<float> getRunningMean();
+
     /**
      * @brief Set the Running Var
      * 
@@ -64,6 +70,12 @@ class BatchNorm2d : public WeightedLayer, public TwoDLayer {
      */
     void setRunningVar(const float* running_mean_input);
 
+    /**
+     * @brief Get the Running Var
+     * 
+     */
+    std::vector<float> getRunningVar();
+
     /**
      * @brief Get output size
      *

include/model/model.hpp

@@ -15,6 +15,8 @@ namespace CUDANet {
 enum TensorType {
     WEIGHT,
     BIAS,
+    RUNNING_MEAN,
+    RUNNING_VAR
 };
 
 struct TensorInfo {