This reverts commit 360eec6910.
This commit is contained in:
1 parent
7d56d45124
commit
a3d5f1d3e3
1 file changed
+2
-2
+2
-2
@@ -5,7 +5,7 @@ Plotting utils
|
|||||||
|
|
||||||
import math
|
import math
|
||||||
import os
|
import os
|
||||||
from copy import deepcopy
|
from copy import copy
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
|
|
||||||
import cv2
|
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=''):
|
def plot_lr_scheduler(optimizer, scheduler, epochs=300, save_dir=''):
|
||||||
# Plot LR simulating training for full epochs
|
# Plot LR simulating training for full epochs
|
||||||
optimizer, scheduler = deepcopy(optimizer), deepcopy(scheduler) # do not modify originals
|
optimizer, scheduler = copy(optimizer), copy(scheduler) # do not modify originals
|
||||||
y = []
|
y = []
|
||||||
for _ in range(epochs):
|
for _ in range(epochs):
|
||||||
scheduler.step()
|
scheduler.step()
|
||||||
|
|||||||
Reference in new issue
Block a user