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