Unverified 提交 60fe5444 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

Update train.py (#3099)

上级 525f4f86
...@@ -367,8 +367,6 @@ def train(hyp, opt, device, tb_writer=None): ...@@ -367,8 +367,6 @@ def train(hyp, opt, device, tb_writer=None):
# Write # Write
with open(results_file, 'a') as f: with open(results_file, 'a') as f:
f.write(s + '%10.4g' * 7 % results + '\n') # append metrics, val_loss f.write(s + '%10.4g' * 7 % results + '\n') # append metrics, val_loss
if len(opt.name) and opt.bucket:
os.system('gsutil cp %s gs://%s/results/results%s.txt' % (results_file, opt.bucket, opt.name))
# Log # Log
tags = ['train/box_loss', 'train/obj_loss', 'train/cls_loss', # train loss tags = ['train/box_loss', 'train/obj_loss', 'train/cls_loss', # train loss
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论