torchinfo 参数量和层数输入输出信息

from torchinfo import summary

# 只显示总参数量和总层数
summary(model.net_g, input_size=(1, 3, 224, 224), verbose=0)

# 显示每一层的名称、类型和参数量
summary(model.net_g, input_size=(1, 3, 224, 224), verbose=1)

# 显示每一层的详细信息（默认）
summary(model.net_g, input_size=(1, 3, 224, 224), verbose=2)

结果如下图所示，CAMixerSR模型结构图：