这个是由于你的数据类默认使用cuda类的tensor,但是读数据的时候应该使用cpu类型的tensor,在forward的时候再将数据变为cuda类型tensor。解决方法就是将数据类中cuda类型的tensor改为cpu类型的tensor,在模型forward的时候再变为cuda类型的tensor
pin_memory报错解决:RuntimeError: cannot pin ‘CUDAComplexFloatType‘ only dense CPU tensors can be pinned
最新推荐文章于 2025-02-14 11:43:35 发布