Torch 在模型的中间层停止前向传播

假设我有一个包含 10 个层的模型,我想在测试时运行到随机层数,比如第 8 层。当我想运行到第 8 层时,我不需要第 9 和第 10 层的结果,因此我希望它在第 8 层停止,不额外消耗资源。我该如何做?非常感谢。

点赞
用户7194361
用户7194361

首先,你需要加载你的预训练模型,然后你可以根据这里定义的 Sequential Model 使用 remove function 来删除任何层。

2018-02-18 13:27:58