va1shn9v / PromptIR

PromptIR: Prompting for All-in-One Blind Image Restoration [NeurIPS 2023]
https://arxiv.org/abs/2306.13090
Other
322 stars 23 forks source link

error #5

Closed zhenyuan1234 closed 1 year ago

zhenyuan1234 commented 1 year ago

Hi, I'm having the following problem with my training, can you tell me how to solve it?

TypeError: lr_scheduler_step() takes 2 positional arguments but 3 were given