问题描述
在官方example中,指标和损失函数均被硬编码。我想知道是否可以在config jsonnet中传递它们,以便可以在具有不同指标的不同数据集中重用我的模型。
解决方法
我知道我以前曾经看过这个问题。从GitHub复制并粘贴:
Metric
是可注册的,因此您可以轻松地向类型为List[Metric]
的模型添加参数,然后在Jsonnet中指定指标。您必须确保这些指标采用完全相同的输入。
对于损失,这有点困难。您将创建自己的Registrable
基类,然后以这种方式实现您要使用的损失。您可以使用Metric
类作为执行此操作的示例。这会有点打字工作,但并不困难。