Revert "Update `plot_lr_scheduler()` (#5864)" (#5920)
Browse filesThis reverts commit 360eec69101ad9ffada78326b715b724b9b7eb0d.
- utils/plots.py +2 -2
utils/plots.py
CHANGED
@@ -5,7 +5,7 @@ Plotting utils
|
|
5 |
|
6 |
import math
|
7 |
import os
|
8 |
-
from copy import
|
9 |
from pathlib import Path
|
10 |
|
11 |
import cv2
|
@@ -243,7 +243,7 @@ def plot_images(images, targets, paths=None, fname='images.jpg', names=None, max
|
|
243 |
|
244 |
def plot_lr_scheduler(optimizer, scheduler, epochs=300, save_dir=''):
|
245 |
# Plot LR simulating training for full epochs
|
246 |
-
optimizer, scheduler =
|
247 |
y = []
|
248 |
for _ in range(epochs):
|
249 |
scheduler.step()
|
|
|
5 |
|
6 |
import math
|
7 |
import os
|
8 |
+
from copy import copy
|
9 |
from pathlib import Path
|
10 |
|
11 |
import cv2
|
|
|
243 |
|
244 |
def plot_lr_scheduler(optimizer, scheduler, epochs=300, save_dir=''):
|
245 |
# Plot LR simulating training for full epochs
|
246 |
+
optimizer, scheduler = copy(optimizer), copy(scheduler) # do not modify originals
|
247 |
y = []
|
248 |
for _ in range(epochs):
|
249 |
scheduler.step()
|