Save model by default (#178)

* Save model by default * Update train.py
2019-03-31 13:11:13 -04:00 · 2019-03-31 13:11:13 -04:00 · 8901e96a38
parent 6b828f184e
commit 8901e96a38
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -170,7 +170,7 @@ def train(
            best_loss = mloss['total']

        # Save training results
-        save = False
+        save = True
        if save:
            # Save latest checkpoint
            checkpoint = {'epoch': epoch,