pytorch:argument 0 is not a Variable

运行sun_test.py时出现如下错误:

出现此错误的原因是把Tensor类型的数据输入训练的模型,需要将输入数据转换为Variable类型,然后可以代码可以正常运行。

猜你喜欢

转载自blog.csdn.net/mywiziqi/article/details/82820286