!2586 [武汉理工大学][高校贡献][Mindspore][Transformer-XL]-MSP GPU单卡和多卡训练脚本
Merge pull request !2586 from XZFFF/master
Showing
- research/nlp/transformer_xl/README.md 298 additions, 0 deletionsresearch/nlp/transformer_xl/README.md
- research/nlp/transformer_xl/README_CN.md 296 additions, 0 deletionsresearch/nlp/transformer_xl/README_CN.md
- research/nlp/transformer_xl/eval.py 77 additions, 0 deletionsresearch/nlp/transformer_xl/eval.py
- research/nlp/transformer_xl/getdata.sh 43 additions, 0 deletionsresearch/nlp/transformer_xl/getdata.sh
- research/nlp/transformer_xl/prep_text8.py 46 additions, 0 deletionsresearch/nlp/transformer_xl/prep_text8.py
- research/nlp/transformer_xl/requirements.txt 3 additions, 0 deletionsresearch/nlp/transformer_xl/requirements.txt
- research/nlp/transformer_xl/script/run_distribute_train_gpu.sh 63 additions, 0 deletions...rch/nlp/transformer_xl/script/run_distribute_train_gpu.sh
- research/nlp/transformer_xl/script/run_eval_gpu.sh 69 additions, 0 deletionsresearch/nlp/transformer_xl/script/run_eval_gpu.sh
- research/nlp/transformer_xl/script/run_standalone_train_gpu.sh 44 additions, 0 deletions...rch/nlp/transformer_xl/script/run_standalone_train_gpu.sh
- research/nlp/transformer_xl/src/__init__.py 14 additions, 0 deletionsresearch/nlp/transformer_xl/src/__init__.py
- research/nlp/transformer_xl/src/callback/eval.py 86 additions, 0 deletionsresearch/nlp/transformer_xl/src/callback/eval.py
- research/nlp/transformer_xl/src/callback/flag.py 26 additions, 0 deletionsresearch/nlp/transformer_xl/src/callback/flag.py
- research/nlp/transformer_xl/src/callback/log.py 68 additions, 0 deletionsresearch/nlp/transformer_xl/src/callback/log.py
- research/nlp/transformer_xl/src/loss_fn/ProjectedAdaptiveLogSoftmaxLoss.py 95 additions, 0 deletions...sformer_xl/src/loss_fn/ProjectedAdaptiveLogSoftmaxLoss.py
- research/nlp/transformer_xl/src/metric/calc.py 59 additions, 0 deletionsresearch/nlp/transformer_xl/src/metric/calc.py
- research/nlp/transformer_xl/src/model/attn.py 165 additions, 0 deletionsresearch/nlp/transformer_xl/src/model/attn.py
- research/nlp/transformer_xl/src/model/dataset.py 157 additions, 0 deletionsresearch/nlp/transformer_xl/src/model/dataset.py
- research/nlp/transformer_xl/src/model/embedding.py 84 additions, 0 deletionsresearch/nlp/transformer_xl/src/model/embedding.py
- research/nlp/transformer_xl/src/model/layer.py 32 additions, 0 deletionsresearch/nlp/transformer_xl/src/model/layer.py
- research/nlp/transformer_xl/src/model/mem_transformer.py 233 additions, 0 deletionsresearch/nlp/transformer_xl/src/model/mem_transformer.py
Please register or sign in to comment