Commit f6f69d1b authored by Anjith GEORGE's avatar Anjith GEORGE

Small fix

parent 4ae1c375
Pipeline #15602 failed with stages
in 3 minutes and 23 seconds
......@@ -104,6 +104,8 @@ class BatchAutoencoder(Extractor, object):
# Initialize the state of the model:
self.model.load_state_dict(model_state)
# Model is used for evaluation only
self.model.train(False)
#==========================================================================
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment