Extreme Multi-label Classification on Wiki10-31K (Inference Time)
20Inference Time (ms)AttentionXML-1
Evaluation Results
| Method | Links | |
|---|---|---|
| AttentionXML-1GPU=Nvidia V100, batch size=12021.10 | 20 | |
| LightXML-1GPU=Nvidia V100, batch size=12021.10 | 27.1 | |
| XR-Transformer-1GPU=Nvidia V100, batch size=12021.10 | 39.1 | |
| X-Transformer-1GPU=Nvidia V100, batch size=12021.10 | 48.1 |