This commit is contained in:
Glenn Jocher 2019-11-19 12:42:12 -08:00
parent b758b9c76e
commit d9805d2fb6
1 changed files with 2 additions and 0 deletions

View File

@ -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: