查看模型结构
form torchsummary import summary
net=LeNet(class=n)
net.initialize_weight()
summary(net,input_size=(c,h,w))
查看模型结构
form torchsummary import summary
net=LeNet(class=n)
net.initialize_weight()
summary(net,input_size=(c,h,w))