pytorch的模型去掉末端的n层,以n=1为例:

new_model = torch.nn.Sequential( *( list(model.children())[:-1] ) )

ps: 模型如何去掉任意位置的某一层?好像不支持?可能去掉之后,图就改变了,问题比较复杂。但是可以通过在load statedict时选择性的加载有需要的参数层,来间接实现这一目的。