Update plot_lr_scheduler() (ultralytics#5864)

shallow copy modify originals
neuralmagic · Dec 3, 2021 · 360eec6 · 360eec6
1 parent 92a7391
commit 360eec6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/utils/plots.py b/utils/plots.py
@@ -5,7 +5,7 @@
 
 import math
 import os
-from copy import copy
+from copy import deepcopy
 from pathlib import Path
 
 import cv2
@@ -243,7 +243,7 @@ def plot_images(images, targets, paths=None, fname='images.jpg', names=None, max
 
 def plot_lr_scheduler(optimizer, scheduler, epochs=300, save_dir=''):
     # Plot LR simulating training for full epochs
-    optimizer, scheduler = copy(optimizer), copy(scheduler)  # do not modify originals
+    optimizer, scheduler = deepcopy(optimizer), deepcopy(scheduler)  # do not modify originals
     y = []
     for _ in range(epochs):
         scheduler.step()