环境: pytorch 1.7.0 pytorch lightning 1.3.8 SqueezeNet 网络结构: 代码复现: #2017 import torch import pytorch_lightning as pl import torch.