mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

Merge pull request #1502 from infomon/main

Browse Source 
Fix typo of "sched-on-updates" argument
This commit is contained in:
Ross Wightman 2022-10-17 14:06:23 -07:00 committed by GitHub
commit 44cea489fb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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

@ -714,7 +714,7 @@ def main():
elif resume_epoch is not None: elif resume_epoch is not None:
start_epoch = resume_epoch start_epoch = resume_epoch
if lr_scheduler is not None and start_epoch > 0: if lr_scheduler is not None and start_epoch > 0:
if args.step_on_updates: if args.sched_on_updates:
lr_scheduler.step_update(start_epoch * updates_per_epoch) lr_scheduler.step_update(start_epoch * updates_per_epoch)
else: else:
lr_scheduler.step(start_epoch) lr_scheduler.step(start_epoch)