parent
92a7391039
commit
360eec6910
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue