火炬在中间层向前停止模型

Question

假设我有一个10层模型，我想运行直到测试时的随机数层，例如第8层。当我想要运行到第8层时，我不需要第9层和第10层的结果，所以我希望它在第8层停止而无需额外费用。我怎样才能做到这一点？非常感谢。

Answer 1

首先，您需要加载预先训练的模型，然后根据定义的here顺序模型，使用remove function删除任何图层。