Unverified 提交 a3d5f1d3 authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

This reverts commit 360eec69.
上级 7d56d451
...@@ -5,7 +5,7 @@ Plotting utils ...@@ -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 ...@@ -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()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论