pytorch报错系列(1)RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor)
在训练期间,运行如下代码import torchimport torch.nn as nnfrom lenet import LeNet, MyNetfrom torchsummary import summarynet = LeNet(classes=2)net.initialize_weights()summary(net, input_size=(3, 32, 32))报...
原创
2020-03-15 22:55:22 ·
1998 阅读 ·
1 评论