Tony-Y / pytorch_warmup

Learning Rate Warmup in PyTorch
https://tony-y.github.io/pytorch_warmup/
MIT License
392 stars 25 forks source link

no attribute named dampening #7

Closed beegica closed 2 years ago

beegica commented 2 years ago

I tried to run your emnist example and got this error: Traceback (most recent call last): File "main.py", line 163, in main() File "main.py", line 152, in main warmup_scheduler, epoch, history) File "main.py", line 42, in train with warmup_scheduler.dampening(): AttributeError: 'UntunedLinearWarmup' object has no attribute 'dampening'

I can't seem to get any of your warmups to work. Do you have any idea why that might be?

Thanks so much!

Tony-Y commented 2 years ago

You probably use an older version. Please upgrade pytorch_warmup as the following:

pip install -U pytorch_warmup