支持平台与模型
支持平台
- GPU：MTT S5000
- CPU：Intel / AMD
- OS: Ubuntu 22.04.4 LTS
支持模型列表
搜推广经典模型
| 模型 | 推理 | 训练 |
|---|---|---|
| WDL | ✅ | ✅ |
| FNN | ✅ | ✅ |
| PNN | ✅ | ✅ |
| DeepFM | ✅ | ✅ |
| NFM | ✅ | ✅ |
| MLR | ✅ | ✅ |
| SharedBottom | ✅ | ✅ |
特征交叉深化模型
| 模型 | 推理 | 训练 |
|---|---|---|
| DCN | ✅ | ✅ |
| xDeepFM | ✅ | ✅ |
| AutoInt | ✅ | ✅ |
| FGCNN | ✅ | ✅ |
| FiBiNet | ✅ | ✅ |
| FWFM | ✅ | ✅ |
用户行为序列建模及多任务模型
| 模型 | 推理 | 训练 |
|---|---|---|
| DIN | ✅ | ✅ |
| DIEN | ✅ | ✅ |
| DSIN | ✅ | ✅ |
| BST | ✅ | ✅ |
| MMoE | ✅ | ✅ |
| ESMM | ✅ | ✅ |
| PLE | ✅ | ✅ |
前沿新架构模型
| 模型 | 推理 | 训练 |
|---|---|---|
| EDCN | ✅ | ✅ |
| DCNmix | ✅ | ✅ |
| Wukong | ✅ | ✅ |
| RankMixer | ✅ | ✅ |
| OneTrans | ✅ | ✅ |
| TokenMixerLarge | ✅ | ✅ |
后续更多模型列表见：https://gitee.com/mthreadsacademy/tensorflow_musa_playground
