W&B log epoch (#1946)
* W&B log epoch * capitalize * W&B log epoch * capitalize * Update train.py New try using https://docs.wandb.ai/library/log#incremental-logging * Update train.py * Update test.py * Update train.py * Update plots.py * Update train.py * Update train.py * label plot step -1 * update * update * update * update * update * update * Update train.py * Update train.py
This commit is contained in:
1 parent
08d3119e09
commit
59c21c7bcb
3 files changed
+4
-4
No files matched your search
@@ -239,8 +239,8 @@ def test(data,
|
||||
if plots:
|
||||
confusion_matrix.plot(save_dir=save_dir, names=list(names.values()))
|
||||
if wandb and wandb.run:
|
||||
wandb.log({"Images": wandb_images})
|
||||
wandb.log({"Validation": [wandb.Image(str(f), caption=f.name) for f in sorted(save_dir.glob('test*.jpg'))]})
|
||||
val_batches = [wandb.Image(str(f), caption=f.name) for f in sorted(save_dir.glob('test*.jpg'))]
|
||||
wandb.log({"Images": wandb_images, "Validation": val_batches}, commit=False)
|
||||
|
||||
# Save JSON
|
||||
if save_json and len(jdict):
|
||||
|
||||
Reference in new issue
Block a user