如何在预训练的 EfficientNetB0 模型中添加额外的层(在特征模块内)?

请您指导我如何在 model.features[5:] 之后添加一个多头注意力层。我想在 features 中添加这个层。