New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

can not train #29

Open

Yintiankuaile666 opened this issue Mar 30, 2023 · 3 comments

Yintiankuaile666 commented Mar 30, 2023 •

edited

Loading

ValueError: The provided lr scheduler "<torch.optim.lr_scheduler.StepLR object at 0x7f5c0c38e6d0>" is invalid

luojin commented Jun 30, 2023

@Yintiankuaile666 do you solve it?

ssh-keyz commented Oct 4, 2023

Having this issue as well

ssh-keyz commented Oct 4, 2023

Solved by downgrading torch and torch vision

pip install torchvision==0.14.1
pip install torch==1.13.1

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment