out of index error when testing with CULane dataset

[NickNameInvalid]

Tried on both culane_res18 and culane_res34 both two models got this out of index error in clr_head

使用预训练culane模型进行验证出现out of index 错误
pytorch=1.8 torchvision=0.9 with cuda version 11.1

[anhuidinglieqiaodaima]

请问这个问题解决了嘛？期待你的回复

[NickNameInvalid]

请问这个问题解决了嘛？期待你的回复

你好 感谢回复！这个问题已经通过降低torch版本解决了 应该是依赖库的版本问题