Inference Time on Extreme Multi-label Classification on Amazon-670K
23.3Inference Time (ms/sample)LightXML-1
Evaluation Results
| Method | Links | |
|---|---|---|
| LightXML-1GPU=Nvidia V100, batch size=12021.10 | 23.3 | |
| XR-Transformer-1GPU=Nvidia V100, batch size=12021.10 | 30.9 | |
| X-Transformer-1GPU=Nvidia V100, batch size=12021.10 | 48 | |
| AttentionXML-1GPU=8 Nvidia V100, batch size=1, model parallel=true2021.10 | 76 |