!2101 MMOE模型实现GPU环境的适配
Merge pull request !2101 from wqx/mmoe
No related branches found
No related tags found
Showing
- research/recommend/mmoe/README_CN.md 93 additions, 35 deletionsresearch/recommend/mmoe/README_CN.md
- research/recommend/mmoe/default_config_gpu.yaml 63 additions, 0 deletionsresearch/recommend/mmoe/default_config_gpu.yaml
- research/recommend/mmoe/eval.py 31 additions, 24 deletionsresearch/recommend/mmoe/eval.py
- research/recommend/mmoe/scripts/run_standalone_eval_gpu.sh 29 additions, 0 deletionsresearch/recommend/mmoe/scripts/run_standalone_eval_gpu.sh
- research/recommend/mmoe/scripts/run_standalone_train_gpu.sh 40 additions, 0 deletionsresearch/recommend/mmoe/scripts/run_standalone_train_gpu.sh
- research/recommend/mmoe/src/callback.py 117 additions, 0 deletionsresearch/recommend/mmoe/src/callback.py
- research/recommend/mmoe/src/data.py 27 additions, 10 deletionsresearch/recommend/mmoe/src/data.py
- research/recommend/mmoe/src/load_dataset.py 20 additions, 14 deletionsresearch/recommend/mmoe/src/load_dataset.py
- research/recommend/mmoe/src/mmoe.py 38 additions, 17 deletionsresearch/recommend/mmoe/src/mmoe.py
- research/recommend/mmoe/src/mmoe_utils.py 29 additions, 10 deletionsresearch/recommend/mmoe/src/mmoe_utils.py
- research/recommend/mmoe/train.py 31 additions, 19 deletionsresearch/recommend/mmoe/train.py
Please register or sign in to comment