NeROIC
NeROIC copied to clipboard
can not train
ValueError: The provided lr scheduler "<torch.optim.lr_scheduler.StepLR object at 0x7f5c0c38e6d0>" is invalid
@Yintiankuaile666 do you solve it?
Having this issue as well
Solved by downgrading torch and torch vision
pip install torchvision==0.14.1
pip install torch==1.13.1