虽然不知道原理,但是看到collate.py或者runtimeerror中提示类型错误,可以去dataset的getitem看看返回数据的数据类型,数据类型需要是tensor,如果不是tensor,那么dataloader可能出现这类错误
site-packages/torch/utils/data/_utils/collate.py中报runtimeerror可能是什么原因?
最新推荐文章于 2024-10-01 20:17:55 发布
虽然不知道原理,但是看到collate.py或者runtimeerror中提示类型错误,可以去dataset的getitem看看返回数据的数据类型,数据类型需要是tensor,如果不是tensor,那么dataloader可能出现这类错误