Fix `--evolve --bucket gs://...` (#6698)
Browse files
train.py
CHANGED
@@ -582,7 +582,7 @@ def main(opt, callbacks=Callbacks()):
|
|
582 |
# ei = [isinstance(x, (int, float)) for x in hyp.values()] # evolvable indices
|
583 |
evolve_yaml, evolve_csv = save_dir / 'hyp_evolve.yaml', save_dir / 'evolve.csv'
|
584 |
if opt.bucket:
|
585 |
-
os.system(f'gsutil cp gs://{opt.bucket}/evolve.csv {
|
586 |
|
587 |
for _ in range(opt.evolve): # generations to evolve
|
588 |
if evolve_csv.exists(): # if evolve.csv exists: select best hyps and mutate
|
|
|
582 |
# ei = [isinstance(x, (int, float)) for x in hyp.values()] # evolvable indices
|
583 |
evolve_yaml, evolve_csv = save_dir / 'hyp_evolve.yaml', save_dir / 'evolve.csv'
|
584 |
if opt.bucket:
|
585 |
+
os.system(f'gsutil cp gs://{opt.bucket}/evolve.csv {evolve_csv}') # download evolve.csv if exists
|
586 |
|
587 |
for _ in range(opt.evolve): # generations to evolve
|
588 |
if evolve_csv.exists(): # if evolve.csv exists: select best hyps and mutate
|