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])