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

Question

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

Answer 1

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