在训练期间,运行如下代码
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=
在训练期间,运行如下代码
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=