From 8901e96a38ee7fc6c832eb97a40c028bb18af763 Mon Sep 17 00:00:00 2001 From: Gabriel Bianconi Date: Sun, 31 Mar 2019 13:11:13 -0400 Subject: [PATCH] Save model by default (#178) * Save model by default * Update train.py --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index c714bc27..da102e4d 100644 --- 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,