updates
This commit is contained in:
parent
81e5514f4f
commit
7997be8bba
2
train.py
2
train.py
|
@ -340,7 +340,7 @@ def train():
|
||||||
|
|
||||||
# Save last checkpoint
|
# Save last checkpoint
|
||||||
torch.save(chkpt, last)
|
torch.save(chkpt, last)
|
||||||
if opt.bucket:
|
if opt.bucket and not opt.evolve:
|
||||||
os.system('gsutil cp %s gs://%s' % (last, opt.bucket)) # upload to bucket
|
os.system('gsutil cp %s gs://%s' % (last, opt.bucket)) # upload to bucket
|
||||||
|
|
||||||
# Save best checkpoint
|
# Save best checkpoint
|
||||||
|
|
Loading…
Reference in New Issue