pytorch/ignite

Tracking the metrics with performance gaps on XLA

Fechada

#2.109 aberto em 14 de jul. de 2021

 (2 comentários) (0 reação) (0 responsável)Python (602 forks)batch import
HacktoberfestPyDataGlobalenhancementhelp wantedmodule: metrics

Métricas do repositório

Stars
 (4.313 estrelas)
Métricas de merge de PR
 (Mesclagem média 18d 23h) (19 fundiu PRs em 30d)

Description

🚀 Feature

Since this commit, the metrics have been disabled with xla device due to a performance regression.

https://github.com/pytorch/ignite/blob/7bdf92322a15ba0bea71ad76f3f785b7ea5907c3/ignite/metrics/metric.py#L224-L225

To solve this, it would be interesting to track which metrics work fine and which have lack of performance. Thus, we could think about more specific enable / disable process.

Guia do colaborador