Example of distributed dataparallel training in PyTorch.
-
Preprocess
python3 dataset.py -
Training
python3 -m torch.distributed.launch --nproc_per_node=2 train.py -
Evaluate
python3 -m torch.distributed.launch --nproc_per_node=1 inference.py
-
Notifications
You must be signed in to change notification settings - Fork 0
Example of distributed dataparallel training in PyTorch.
License
tmyok/pytorch_DDP_example
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Example of distributed dataparallel training in PyTorch.
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published