weixin_44776845的博客

在训练期间，运行如下代码 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)) 报...