配置中的传递损失函数和度量



在官方示例中,度量和损失函数都是硬编码的。我想知道我们是否可以在配置jsonnet中传递这些,这样我就可以在具有不同度量的不同数据集中重用我的模型。

我知道我以前看过那个问题。从GitHub:复制和粘贴

Metric是可注册的,因此您可以轻松地向List[Metric]类型的模型添加参数,然后在Jsonnet中指定度量。您必须确保这些度量采用完全相同的输入。

对于损失来说,这有点困难。您将创建自己的Registrable基类,然后以这种方式实现想要使用的损失。您可以使用Metric类作为如何执行此操作的示例。这将是一项打字工作,但并不困难。

最新更新