Kaynağa Gözat

W&B mosaic log bug fix (#1949)

5.0
Glenn Jocher GitHub 3 yıl önce
ebeveyn
işleme
f4a78e1b40
Veri tabanında bu imza için bilinen anahtar bulunamadı GPC Anahtar Kimliği: 4AEE18F83AFDEB23
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  1. +1
    -1
      train.py

+ 1
- 1
train.py Dosyayı Görüntüle

@@ -318,7 +318,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
# if tb_writer:
# tb_writer.add_image(f, result, dataformats='HWC', global_step=epoch)
# tb_writer.add_graph(model, imgs) # add model to tensorboard
elif plots and ni > 3 and wandb:
elif plots and ni == 10 and wandb:
wandb.log({"Mosaics": [wandb.Image(str(x), caption=x.name) for x in save_dir.glob('train*.jpg')
if x.exists()]})


Yükleniyor…
İptal
Kaydet