通过torchsummary查看网络

$$通过torchsummary查看网络$$

from torchsummary import summary
import major_config
summary(major_config.model,input_size=(3,32,32),device="cpu")

import major_config
from torchsummary import summary
net = major_config.model
summary(net, input_size=major_config.input_size, device="cpu")