Fix val `plots=plots` (#7524)

2022-04-21 18:21:01 -07:00 · 2022-04-21 18:21:01 -07:00 · 23718df1c6
parent 6ea81bb3a9
commit 23718df1c6
1 changed files with 1 additions and 1 deletions
--- a/train.py
+++ b/train.py
@ -461,7 +461,7 @@ def train(hyp, opt, device, callbacks):  # hyp is path/to/hyp.yaml or hyp dictio
                        save_dir=save_dir,
                        save_json=is_coco,
                        verbose=True,
-                        plots=True,
+                        plots=plots,
                        callbacks=callbacks,
                        compute_loss=compute_loss)  # val best model with plots
                    if is_coco: