From 110f5941b7d9e736443c7c9c302cfe904feed62c Mon Sep 17 00:00:00 2001 From: Chaoyue Wang Date: Wed, 19 Jul 2017 21:33:41 -0400 Subject: [PATCH] Fixed bn layer param order issue --- pspnet.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/pspnet.py b/pspnet.py index e50c396..2374e9a 100644 --- a/pspnet.py +++ b/pspnet.py @@ -29,10 +29,10 @@ def set_weights(model, weights): # mean *= scale # variance *= scale - model.get_layer(layer.name).set_weights([mean, variance, - scale, offset]) - # model.get_layer(layer.name).set_weights([scale, offset, - # mean, variance]) + # model.get_layer(layer.name).set_weights([mean, variance, + # scale, offset]) + model.get_layer(layer.name).set_weights([scale, offset, + mean, variance]) # model.get_layer(layer.name).set_weights([scale, offset, # mean, variance])