Compute loss on final val (#5017)

2021-09-30 15:16:11 -07:00 · 2021-09-30 15:16:11 -07:00 · 89a6e9ffa7
parent b20e381492
commit 89a6e9ffa7
1 changed files with 2 additions and 1 deletions
--- 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)}")