Commit e23948da authored by Anjali Sridhar's avatar Anjali Sridhar Committed by TensorFlower Gardener
Browse files

For models running in Eager mode, do not update the weights of the BatchNorm... 

For models running in Eager mode, do not update the weights of the BatchNorm layer if the layer's trainable argument is False.

This change is required in Eager mode to freeze a layer's weights when we set the layer's trainable attribute to False.

This should not be confused with the "training" attribute which refers to a model's training or inference mode behavior.

PiperOrigin-RevId: 185625661
parent 7325919e
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment