pytorch删除模型model中的某些层(或模块)

核心思想: 写一个空的层 x= nn.Sequential();

                   然后替换原model中的相应部分

model=ConvNext(num_class=1000)
print(model)
remove_head=nn.Sequential()
model.head=remove_head
print(model)

你可能感兴趣的:(pytorch,深度学习,人工智能)