pytorch/ignite

Tracking the metrics with performance gaps on XLA

Fermée

#2 109 ouverte le 14 juil. 2021

 (2 commentaires) (0 réaction) (0 personne assignée)Python (602 forks)batch import
HacktoberfestPyDataGlobalenhancementhelp wantedmodule: metrics

Métriques du dépôt

Stars
 (4 313 étoiles)
Métriques de merge PR
 (Merge moyen 18j 23h) (19 PRs mergées en 30 j)

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.

Guide contributeur