pytorch/ignite

More EpochMetric's compute_fn output types

Open

#1,757 opened on 2021年3月8日

GitHub で見る
 (3 comments) (0 reactions) (0 assignees)Python (4,313 stars) (602 forks)batch import
HacktoberfestPyDataGlobalenhancementhelp wantedmodule: metrics

説明

🚀 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.

コントリビューターガイド

技術スタック
pythonpytorch
領域
machine learning
Issue 種別
feature
難度新しい貢献者にとっての実装難度です。1 は非常に小さな変更、5 は専門的な作業です。
3
推定時間経験ある貢献者が調査、実装、テスト、pull request 準備にかけるおおよその時間範囲です。
1-2 days
活動状況Issue が今どれだけ取り組みやすいかを示します。新しい、活発、古い、ブロック中、またはメンテナー入力待ちなどです。
stale
明確さ期待される変更、受け入れ条件、次の手順がどれだけ明確かを示します。
mostly clear
前提条件
Familiarity with PyTorchUnderstanding of EpochMetric class
初心者向け度初回貢献者にどれだけ取り組みやすいかを 1-100 で推定したスコアです。
60
調査方針
Review the existing implementation of EpochMetric in the ignite codebase (likely in ignite/metrics/epoch metric.py). Understand how compute fn is currently used and the output types it supports. Examine the code snippet in the linked comment for a potential approach. Then implement support for scalars, tensors, and tuple/list/mapping of tensors. Also add a more user friendly error message for unsupported types. Finally, update the documentation to reflect the new capabilities.