本文共 1157 字，大约阅读时间需要 3 分钟。

������ torchsummary ������������������ PyTorch ������������

���PyTorch��������� torchsummary ������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������ torchsummary ��������������� PyTorch ������������������������

������������������������������������������������������������������������������ torchsummary ���������������������������

from torchsummary import summary
import major_config

���������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������

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

������������������������������������������������������������������������������������������������������

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

���������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������������