can not train

[Yintiankuaile666]

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

[luojin]

@Yintiankuaile666 do you solve it?

[deepc0py]

Having this issue as well

[deepc0py]

Solved by downgrading torch and torch vision

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