Extreme Multi-label Classification on Wiki 500K (Inference Time)
27.3Inference Time (ms/sample)LightXML-1
Evaluation Results
| Method | Links | |
|---|---|---|
| LightXML-1GPU=Nvidia V100, batch size=12021.10 | 27.3 | |
| XR-Transformer-1GPU=Nvidia V100, batch size=12021.10 | 33.9 | |
| X-Transformer-1GPU=Nvidia V100, batch size=12021.10 | 48.1 | |
| AttentionXML-1GPU=8 Nvidia V100, batch size=1, model parallel=true2021.10 | 80.1 |