pytorch/ignite

Tracking the metrics with performance gaps on XLA

Chiusa

#2109 aperta il 14 lug 2021

 (2 commenti) (0 reazioni) (0 assegnatari)Python (602 fork)batch import
HacktoberfestPyDataGlobalenhancementhelp wantedmodule: metrics

Metriche repository

Star
 (4313 stelle)
Metriche merge PR
 (Merge medio 18g 23h) (19 PR mergiate in 30 g)

Descrizione

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

Guida contributor