diff --git a/train.py b/train.py index 4833eebf..325d32ac 100644 --- a/train.py +++ b/train.py @@ -317,6 +317,8 @@ def train(): # Write epoch results with open(results_file, 'a') as f: f.write(s + '%10.3g' * 7 % results + '\n') # P, R, mAP, F1, test_losses=(GIoU, obj, cls) + if len(opt.name) and opt.bucket and not opt.prebias: + os.system('gsutil cp %s gs://%s' % (results_file, opt.bucket)) # Write Tensorboard results if tb_writer: