File size: 256 Bytes
1ab03a3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
#!/bin/sh
#SBATCH --gres=gpu:4
#SBATCH --cpus-per-gpu=8
#SBATCH --mem-per-gpu=32GB
#SBATCH --nodes=2
#SBATCH --ntasks=8
#SBATCH --partition=gpu-cluster
#SBATCH --time=72:00:00
export NCCL_DEBUG=INFO
export PYTHONFAULTHANDLER=1
srun python run_ffhq256.py |