torch-ddp
This is a project of data parallel that running on NLP tasks.
You can e-mail Jiamao at [email protected] , if you have any questions.
Requirements
Our code works with the following environment.
python>=3.7pytorch>=1.3transformersspacy
Tips
you should contain zh_core_web_sm belonged spacy when you running the project.
Run
0:python pre_myfile.py 1:python generate.py 2:python data__process.py 3:python python -m torch.distributed.launch --nproc_per_node=1 --use_env train.py.py