我已经上传了某个模型
from efficientnet_pytorch import EfficientNet
model = EfficientNet.from_pretrained(model)
Run Code Online (Sandbox Code Playgroud)
我可以看到模型:
print(model.state_dict())
Run Code Online (Sandbox Code Playgroud)
该模型包含相当多的层,我只想获取前 50 层。请告诉我如何做到这一点。