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]>
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', '
|
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()
|