Bootstrap

在fastreid模块中是怎么将 resnet101蒸馏成resnet34,去除模型保存了FC层和优化器等参数的呢...

在 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():
  
;