【问题标题】:Torch stop model forward at intermediate layer在中间层向前的火炬停止模型
【发布时间】:2018-07-28 14:30:39
【问题描述】:

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

【问题讨论】:

    标签: lua neural-network torch


    【解决方案1】:

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

    【讨论】:

    • 谢谢,测试中随机数会发生变化,需要去掉网络动态。
    猜你喜欢
    • 2021-07-04
    • 1970-01-01
    • 2020-09-21
    • 2022-08-04
    • 2020-01-23
    • 1970-01-01
    • 1970-01-01
    • 2022-07-20
    • 1970-01-01
    相关资源
    最近更新 更多