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.

貢獻者指南