瀏覽代碼

Update train.py (#3099)

modifyDataloader
Glenn Jocher GitHub 3 年之前
父節點
當前提交
60fe54449d
沒有發現已知的金鑰在資料庫的簽署中 GPG Key ID: 4AEE18F83AFDEB23
共有 1 個文件被更改,包括 0 次插入2 次删除
  1. +0
    -2
      train.py

+ 0
- 2
train.py 查看文件

@@ -367,8 +367,6 @@ def train(hyp, opt, device, tb_writer=None):
# Write
with open(results_file, 'a') as f:
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
tags = ['train/box_loss', 'train/obj_loss', 'train/cls_loss', # train loss

Loading…
取消
儲存