DDP-example

Distributed training example.

pip install -r requirements.txt

Single GPU training:

CUDA_VISIBLE_DEVICES=0 python train.py

Distributed training using two GPUs:

CUDA_VISIBLE_DEVICES=0,1 python train_ddp.py -g 2

Distributed training using two GPUs with Mixed Precision:

CUDA_VISIBLE_DEVICES=0,1 python train_ddp_mp.py -g 2

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
dataset.py		dataset.py
densenet.py		densenet.py
requirements.txt		requirements.txt
state.py		state.py
train.py		train.py
train_ddp.py		train_ddp.py
train_ddp_mp.py		train_ddp_mp.py

Provide feedback