pytorch报错系列（1）RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)

在训练期间，运行如下代码

import torch
import torch.nn as nn
from lenet import LeNet, MyNet
from torchsummary import summary

net = LeNet(classes=2)
net.initialize_weights()
summary(net, input_size=(3, 32, 32))

报错：
RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)

在torchsummary.summary( )中出现了上述错误
原因：输入类型是GPU（torch.cuda.FloatTensor），而参数类型是CPU（torch.FloatTensor），类型不一致而报错

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
net = net.to(device)

或者

net = net.cuda()

这是将 net 转为 cuda，两者类型一致，问题解决。