python - Pytorch 点燃 create_supervised_evaluator 不返回自定义指标

Question

我正在尝试将 ignite.engine 中的 create_supervised_evaluator 与我的自定义指标一起使用

evaluator = create_supervised_evaluator(model, metrics={'perform': Perform()}, device=device)

但是evaluator.state.metrics.keys()正在返回空状态字典。

我正在使用 PyTorch == 1.7.1 和 pytorch-ignite 0.4.2 。任何人都可以让我知道我哪里出错了。

Answer 1

显然你还没有运行评估器，你必须先这样做

evaluator.run(data_loader, max_epochs=100)

然后这样做将为您返回指标字典

evaluator.state.metrics