updates
This commit is contained in:
parent
ac8d78382a
commit
4b9d73f931
3
train.py
3
train.py
|
@ -437,8 +437,7 @@ if __name__ == '__main__':
|
||||||
train() # train normally
|
train() # train normally
|
||||||
|
|
||||||
else: # Evolve hyperparameters (optional)
|
else: # Evolve hyperparameters (optional)
|
||||||
opt.notest = True # only test final epoch
|
opt.notest, opt.nosave = True, True # only test/save final epoch
|
||||||
opt.nosave = True # only save final checkpoint
|
|
||||||
if opt.bucket:
|
if opt.bucket:
|
||||||
os.system('gsutil cp gs://%s/evolve.txt .' % opt.bucket) # download evolve.txt if exists
|
os.system('gsutil cp gs://%s/evolve.txt .' % opt.bucket) # download evolve.txt if exists
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue