Fixed bn layer param order issue

This commit is contained in:
Chaoyue Wang
2017-07-19 21:33:41 -04:00
parent 27b41dce01
commit 110f5941b7
+4 -4
View File
@@ -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])