问题 可视化网络时,出现下面错误 RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same 解决方法 原因:可视化网络使用的时cpu 将网络转成cuda net = CRNN(num_classes=1000) 修改后 net = CRNN(num_classes=1000) net.cuda()