Revert "Update plot_lr_scheduler() (#5864)" (#5920)

This reverts commit 360eec6910.
This commit is contained in:
Glenn Jocher authored and GitHub committed 2021-12-08 16:46:24 +01:00
1 parent 7d56d45124
commit a3d5f1d3e3
1 file changed
+2 -2
+2 -2
View File
@@ -5,7 +5,7 @@ Plotting utils
import math
import os
from copy import deepcopy
from copy import copy
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 = deepcopy(optimizer), deepcopy(scheduler) # do not modify originals
optimizer, scheduler = copy(optimizer), copy(scheduler) # do not modify originals
y = []
for _ in range(epochs):
scheduler.step()