apoorva/greenhouse
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Save model by default (#178)

Browse Source 
* Save model by default

* Update train.py
This commit is contained in:
Gabriel Bianconi 2019-03-31 13:11:13 -04:00 committed by Glenn Jocher
parent 6b828f184e
commit 8901e96a38
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -170,7 +170,7 @@ def train(
best_loss = mloss['total'] best_loss = mloss['total']
# Save training results # Save training results
save = False save = True
if save: if save:
# Save latest checkpoint # Save latest checkpoint
checkpoint = {'epoch': epoch, checkpoint = {'epoch': epoch,