Commit 4b52908c authored by Frisinghelli Daniel's avatar Frisinghelli Daniel
Browse files

Use correct Early Stopping model.

parent 87e29166
......@@ -1120,7 +1120,8 @@ class NetworkTrainer(BaseConfig):
(0, 'valid_accu', np.max) if self.mode == 'max' else
(np.inf, 'valid_loss', np.min))
else:
self.best, self.metric, self.mfn = (np.inf, 'valid_loss', np.min)
self.mode, self.best, self.metric, self.mfn = (
'min', np.inf, 'valid_loss', np.min)
# initialize metric tracker
self.tracker.initialize()
......
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