Convert the model into training mode will keeping the normalization layer freezed.
(self, mode=True)
source not stored for this graph (policy: none)
no outgoing calls