asim266 glenn-jocher commited on
Commit
9ef4760
·
unverified ·
1 Parent(s): 412b113

Rename precision_recall_curve.png bug fix (#1431)

Browse files

* fix error in train.py line 400, changed precision-recall_curve.png to precision_recall_curve.png and fix error in test.py of {wandb.init() before wandb.log()} by importing wand and initializing it before wand.log()

* Update test.py

Co-authored-by: Glenn Jocher <[email protected]>

Files changed (1) hide show
  1. train.py +1 -1
train.py CHANGED
@@ -397,7 +397,7 @@ def train(hyp, opt, device, tb_writer=None, wandb=None):
397
  plot_results(save_dir=save_dir) # save as results.png
398
  if wandb:
399
  wandb.log({"Results": [wandb.Image(str(save_dir / x), caption=x) for x in
400
- ['results.png', 'precision-recall_curve.png']]})
401
  logger.info('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600))
402
  else:
403
  dist.destroy_process_group()
 
397
  plot_results(save_dir=save_dir) # save as results.png
398
  if wandb:
399
  wandb.log({"Results": [wandb.Image(str(save_dir / x), caption=x) for x in
400
+ ['results.png', 'precision_recall_curve.png']]})
401
  logger.info('%g epochs completed in %.3f hours.\n' % (epoch - start_epoch + 1, (time.time() - t0) / 3600))
402
  else:
403
  dist.destroy_process_group()