Compute loss on final val (#5017)

ultralytics · Sep 30, 2021 · 89a6e9f · 89a6e9f
1 parent b20e381
commit 89a6e9f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -419,7 +419,8 @@ def train(hyp,  # path/to/hyp.yaml or hyp dictionary
                                             save_json=is_coco,
                                             verbose=True,
                                             plots=True,
-                                            callbacks=callbacks)  # val best model with plots
+                                            callbacks=callbacks,
+                                            compute_loss=compute_loss)  # val best model with plots
 
         callbacks.run('on_train_end', last, best, plots, epoch)
         LOGGER.info(f"Results saved to {colorstr('bold', save_dir)}")