updates
This commit is contained in:
parent
b758b9c76e
commit
d9805d2fb6
2
train.py
2
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:
|
||||
|
|
Loading…
Reference in New Issue