【发布时间】:2018-07-28 14:30:39
【问题描述】:
假设我有一个 10 层模型,我想在测试时运行到随机数层,例如第 8 层。当我想运行到第 8 层时,我不需要第 9 层和第 10 层的结果,所以我希望它停止在第 8 层而无需额外费用。我怎样才能做到这一点?非常感谢。
【问题讨论】:
标签: lua neural-network torch
假设我有一个 10 层模型,我想在测试时运行到随机数层,例如第 8 层。当我想运行到第 8 层时,我不需要第 9 层和第 10 层的结果,所以我希望它停止在第 8 层而无需额外费用。我怎样才能做到这一点?非常感谢。
【问题讨论】:
标签: lua neural-network torch
首先,您需要加载您的预训练模型,然后您可以根据定义的序列模型here 使用remove function 删除任何层。
【讨论】: