diff --git a/train.py b/train.py index c774430df..f6e66cb0e 100644 --- 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: