W&B: fix refactor bugs (#4069)

This commit is contained in:
Ayush Chaurasia authored and GitHub committed 2021-07-19 13:57:13 +02:00
1 parent 7fdcc77bf4
commit 0cc7c58787
2 files changed
+5 -5

No files matched your search

+1 -1
View File
@@ -215,7 +215,7 @@ def run(data,
save_one_txt(predn, save_conf, shape, file=save_dir / 'labels' / (path.stem + '.txt'))
if save_json:
save_one_json(predn, jdict, path, class_map) # append to COCO-JSON dictionary
if wandb_logger:
if wandb_logger and wandb_logger.wandb_run:
wandb_logger.val_one_image(pred, predn, path, names, img[si])
# Plot images