NeROIC icon indicating copy to clipboard operation
NeROIC copied to clipboard

can not train

Open Yintiankuaile666 opened this issue 2 years ago • 3 comments

image

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

Yintiankuaile666 avatar Mar 30 '23 15:03 Yintiankuaile666

@Yintiankuaile666 do you solve it?

luojin avatar Jun 30 '23 12:06 luojin

Having this issue as well

deepc0py avatar Oct 04 '23 05:10 deepc0py

Solved by downgrading torch and torch vision

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

deepc0py avatar Oct 04 '23 05:10 deepc0py