在 fastai 模块中,可以使用以下代码将 ResNet101 蒸馏成 ResNet34:
from fastai.vision.models import resnet34# 加载 ResNet101 模型
model = resnet101()
# 从 ResNet101 模型中复制参数到 ResNet34 模型
model_34 = resnet34()
model_34.load_state_dict(model.state_dict())
# 去除模型中的全连接层和优化器参数
for param in model_34.parameters():