Run with single node multi GPU

torchrun --standalone --nnodes=1 --nproc_per_node=gpu train.py

Name		Name	Last commit message	Last commit date
Latest commit History 14 Commits
dataset		dataset
loss		loss
optim		optim
.gitignore		.gitignore
README.md		README.md
config.py		config.py
logger.py		logger.py
train.py		train.py
utils.py		utils.py

Provide feedback