pytorch/ignite

More EpochMetric's compute_fn output types

开放

#1,757 创建于 2021年3月8日

 (5 条评论) (0 个反应) (0 位负责人)Python (602 个派生)batch import
HacktoberfestPyDataGlobalenhancementhelp wantedmodule: metrics

仓库指标

星标
 (4,313 个星标)
PR 合并指标
 (PR 指标待抓取)

描述

🚀 Feature

Following https://github.com/pytorch/ignite/pull/1700#issuecomment-787151203 we would like to improve EpochMetric by supporting more output types returned by compute_fn:

  • EpochMetric should work with compute_fn outputs: scalars, tensors, tuple/list/mapping of tensors vs currently only scalars

A code snippet on possible implementation: https://github.com/pytorch/ignite/pull/1700#issuecomment-790413236 (It is not tested and maybe wrong, just an idea of how to do it).

It would be also helpful to raise a more user friendly error message if output type is not supported.

We need to update the docs as well.

If interested in fixing this issue, please leave a comment to assign the issue to you.

贡献者指南