glenn-jocher commited on
Commit
89a6e9f
1 Parent(s): b20e381

Compute loss on final val (#5017)

Browse files
Files changed (1) hide show
  1. train.py +2 -1
train.py CHANGED
@@ -419,7 +419,8 @@ def train(hyp, # path/to/hyp.yaml or hyp dictionary
419
  save_json=is_coco,
420
  verbose=True,
421
  plots=True,
422
- callbacks=callbacks) # val best model with plots
 
423
 
424
  callbacks.run('on_train_end', last, best, plots, epoch)
425
  LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}")
 
419
  save_json=is_coco,
420
  verbose=True,
421
  plots=True,
422
+ callbacks=callbacks,
423
+ compute_loss=compute_loss) # val best model with plots
424
 
425
  callbacks.run('on_train_end', last, best, plots, epoch)
426
  LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}")