Unverified 提交 360eec69 authored 作者: Can's avatar Can 提交者: GitHub

Update `plot_lr_scheduler()` (#5864)

shallow copy modify originals
上级 92a73910
......@@ -5,7 +5,7 @@ Plotting utils
import math
import os
from copy import copy
from copy import deepcopy
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 = copy(optimizer), copy(scheduler) # do not modify originals
optimizer, scheduler = deepcopy(optimizer), deepcopy(scheduler) # do not modify originals
y = []
for _ in range(epochs):
scheduler.step()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论