Xsmos commited on
Commit
da322c6
·
verified ·
1 Parent(s): 3fd64a1
Files changed (2) hide show
  1. .gitignore +2 -0
  2. perlmutter_diffusion.sbatch +1 -1
.gitignore CHANGED
@@ -12,3 +12,5 @@ data/
12
  # phoenix*
13
  # frontera*
14
  error-*
 
 
 
12
  # phoenix*
13
  # frontera*
14
  error-*
15
+ logs/
16
+
perlmutter_diffusion.sbatch CHANGED
@@ -38,6 +38,6 @@ srun python diffusion.py \
38
  --train "$SCRATCH/LEN128-DIM64-CUB16-Tvir[4, 6]-zeta[10, 250]-0809-123640.h5" \
39
  --num_new_img_per_gpu 50 \
40
  --max_num_img_per_gpu 2 \
41
- --gradient_accumulation_steps 30 \
42
  #--resume outputs/model-N2000-device_count1-node8-epoch19-19004529 \
43
 
 
38
  --train "$SCRATCH/LEN128-DIM64-CUB16-Tvir[4, 6]-zeta[10, 250]-0809-123640.h5" \
39
  --num_new_img_per_gpu 50 \
40
  --max_num_img_per_gpu 2 \
41
+ --gradient_accumulation_steps 8 \
42
  #--resume outputs/model-N2000-device_count1-node8-epoch19-19004529 \
43