错误原因:网络的输入和网络没有全部放在cuda()上。
解决办法:
举例如下:
注意看图中圈起来部分,这时输入是已经放在cuda()上了,而网络net_feature没有放在cuda()上,所以会报上面的错误。这时只要再把网络放到cuda()上就可以了。然后就不会再报错了。
错误原因:网络的输入和网络没有全部放在cuda()上。
解决办法:
举例如下:
注意看图中圈起来部分,这时输入是已经放在cuda()上了,而网络net_feature没有放在cuda()上,所以会报上面的错误。这时只要再把网络放到cuda()上就可以了。然后就不会再报错了。