From f7e075fe930f3c083d07f48cdc6ee7de02f34894 Mon Sep 17 00:00:00 2001 From: Glenn Jocher Date: Sat, 22 Aug 2020 15:47:39 -0700 Subject: [PATCH] gsutil cp yaml when evolving to GCP bucket --- utils/general.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/utils/general.py b/utils/general.py index 83509fb..306aae0 100755 --- a/utils/general.py +++ b/utils/general.py @@ -866,9 +866,6 @@ def print_mutation(hyp, results, yaml_file='hyp_evolved.yaml', bucket=''): x = x[np.argsort(-fitness(x))] # sort np.savetxt('evolve.txt', x, '%10.3g') # save sort by fitness - if bucket: - os.system('gsutil cp evolve.txt gs://%s' % bucket) # upload evolve.txt - # Save yaml for i, k in enumerate(hyp.keys()): hyp[k] = float(x[0, i + 7]) @@ -878,6 +875,9 @@ def print_mutation(hyp, results, yaml_file='hyp_evolved.yaml', bucket=''): f.write('# Hyperparameter Evolution Results\n# Generations: %g\n# Metrics: ' % len(x) + c + '\n\n') yaml.dump(hyp, f, sort_keys=False) + if bucket: + os.system('gsutil cp evolve.txt %s gs://%s' % (yaml_file, bucket)) # upload + def apply_classifier(x, model, img, im0): # applies a second stage classifier to yolo outputs @@ -1273,4 +1273,3 @@ def plot_results(start=0, stop=0, bucket='', id=(), labels=(), fig.tight_layout() ax[1].legend() fig.savefig(Path(save_dir) / 'results.png', dpi=200) -