huggingface / huggingface/pytorch-image-models
[BUG] Training job cannot resume if LR scheduler is Plateau.
Open
bug
- Dominant language
- Python
- Stars
- 37.1k
- Forks
- 5.2k
- Avg merge
- 1d 13h
- Merged PRs (30d)
- 34
Description
File "Trainer.py", line 233, in main
lr_scheduler.step(start_epoch)
File "/multimedia-nfs/wuye/libs/miniconda3/envs/py38/lib/python3.8/site-packages/timm/scheduler/plateau_lr.py", line 83, in step
self.lr_scheduler.step(metric, epoch) # step the base scheduler
File "/multimedia-nfs/wuye/libs/miniconda3/envs/py38/lib/python3.8/site-packages/torch/optim/lr_scheduler.py", line 624, in step
current = float(metrics)
TypeError: float() argument must be a string or a number, not 'NoneType'
Trainer.py is just renamed from official training script.
Contributor guide
Assessment
This issue has not been assessed yet.