zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

disable LR plot to suppress warning message

This commit is contained in:
Glenn Jocher 2020-07-09 15:16:57 -07:00
parent 24c5a941f0
commit 72d5b58b9a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -108,7 +108,7 @@ def train(hyp):
# Scheduler https://arxiv.org/pdf/1812.01187.pdf # Scheduler https://arxiv.org/pdf/1812.01187.pdf
lf = lambda x: (((1 + math.cos(x * math.pi / epochs)) / 2) ** 1.0) * 0.9 + 0.1 # cosine lf = lambda x: (((1 + math.cos(x * math.pi / epochs)) / 2) ** 1.0) * 0.9 + 0.1 # cosine
scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf) scheduler = lr_scheduler.LambdaLR(optimizer, lr_lambda=lf)
plot_lr_scheduler(optimizer, scheduler, epochs, save_dir=log_dir) # plot_lr_scheduler(optimizer, scheduler, epochs, save_dir=log_dir)
# Load Model # Load Model
google_utils.attempt_download(weights) google_utils.attempt_download(weights)