glenn-jocher commited on
Commit
a3d5f1d
1 Parent(s): 7d56d45

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

Browse files

This reverts commit 360eec69101ad9ffada78326b715b724b9b7eb0d.

Files changed (1) hide show
  1. 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 deepcopy
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 = deepcopy(optimizer), deepcopy(scheduler) # do not modify originals
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()