From 9b63d1b0a7cbcee78e480e2bd3ec11d8fb4247d2 Mon Sep 17 00:00:00 2001 From: Baoguang Shi Date: Mon, 22 Feb 2016 12:18:03 +0800 Subject: [PATCH] Fix compatibility issues with the latest Torch7 version; Please re-download the updated model --- src/utilities.lua | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/utilities.lua b/src/utilities.lua index 597e9ad..8119055 100644 --- a/src/utilities.lua +++ b/src/utilities.lua @@ -224,7 +224,7 @@ end function modelState(model) --[[ Get model state, including model parameters (weights and biases) and - running mean/std in batch normalization layers + running mean/var in batch normalization layers ARGS: - `model` : network model RETURN: @@ -235,12 +235,12 @@ function modelState(model) local bnLayers = model:findModules('nn.BatchNormalization') for i = 1, #bnLayers do bnVars[#bnVars+1] = bnLayers[i].running_mean - bnVars[#bnVars+1] = bnLayers[i].running_std + bnVars[#bnVars+1] = bnLayers[i].running_var end local bnLayers = model:findModules('nn.SpatialBatchNormalization') for i = 1, #bnLayers do bnVars[#bnVars+1] = bnLayers[i].running_mean - bnVars[#bnVars+1] = bnLayers[i].running_std + bnVars[#bnVars+1] = bnLayers[i].running_var end local state = {parameters = parameters, bnVars = bnVars} return state