train.py results.txt to bucket bug fix

5 years ago · 98fc483abc
parent 01a67a9742
commit 98fc483abc
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -303,7 +303,7 @@ def train(hyp):
        with open(results_file, 'a') as f:
            f.write(s + '%10.4g' * 7 % results + '\n')  # P, R, mAP, F1, test_losses=(GIoU, obj, cls)
        if len(opt.name) and opt.bucket:
-            os.system('gsutil cp results.txt gs://%s/results/results%s.txt' % (opt.bucket, opt.name))
+            os.system('gsutil cp %s gs://%s/results/results%s.txt' % (results_file, opt.bucket, opt.name))

        # Tensorboard
        if tb_writer: