pytorch的模型去掉末端的n层，以n=1为例：
new_model = torch.nn.Sequential( *( list(model.children())[:-1] ) )
ps: 模型如何去掉任意位置的某一层？好像不支持？可能去掉之后，图就改变了，问题比较复杂。但是可以通过在load statedict时选择性的加载有需要的参数层，来间接实现这一目的。
pytorch的模型去掉末端的n层，以n=1为例：
new_model = torch.nn.Sequential( *( list(model.children())[:-1] ) )
ps: 模型如何去掉任意位置的某一层？好像不支持？可能去掉之后，图就改变了，问题比较复杂。但是可以通过在load statedict时选择性的加载有需要的参数层，来间接实现这一目的。
名师互学网 版权所有 (c)2021-2022 ICP备案号：晋ICP备2021003244-6号
