1 file changed
+2
-2
+2
-2
@@ -5,7 +5,7 @@ Plotting utils
|
||||
|
||||
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()
|
||||
|
||||
Reference in new issue
Block a user