/Slurm-DDP-Pytorch

A boilerplate repository to help you easily set-up Pytorch DDP training in SLURM clusters.

Primary LanguagePythonMIT LicenseMIT

Slurm-DDP-Pytorch

A boilerplate repository to help you easily set-up Pytorch DDP training in SLURM clusters.

Run

sbatch ddp_launcher.sh