From 756a91e2de8457d547d9661ea2998f3dcdd85e96 Mon Sep 17 00:00:00 2001 From: Can Date: Fri, 3 Dec 2021 20:37:45 +0800 Subject: [PATCH] Update `plot_lr_scheduler()` (#5864) shallow copy modify originals --- utils/plots.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/utils/plots.py b/utils/plots.py index 69037ee9af70..5742d050fdf5 100644 --- 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()