mirror of
https://github.com/wassname/PSPNet-Keras-tensorflow.git
synced 2026-09-09 11:15:19 +08:00
Fixed bn layer param order issue
This commit is contained in:
@@ -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])
|
||||
|
||||
|
||||
Reference in New Issue
Block a user