0725-2238

.gitignore

@@ -3,7 +3,6 @@ Report-*
 *.h5
 __pycache__/
 hive*
-phoenix*
 *pace.gatech.edu
 frontera*
 test*
@@ -11,3 +10,4 @@ ST4Diffusion/
 runs/
 outputs/
 data/
+# phoenix*

diffusion.py

@@ -249,7 +249,7 @@ class TrainConfig:
     stride = (2,4) if dim == 2 else (2,2,2)
     num_image = 30#60#6000#1000#2000#20000#15000#7000#25600#3000#10000#1000#10000#5000#2560#800#2560
     batch_size = 10#50#20#50#1#2#50#20#2#100 # 10
-    n_epoch = 50#100#50#100#30#120#5#4# 10#50#20#20#2#5#25 # 120
+    n_epoch = 5#50#100#50#100#30#120#5#4# 10#50#20#20#2#5#25 # 120
     HII_DIM = 64
     num_redshift = 512#64#512#64#256CUDAoom#128#64#512#128#64#512#256#256#64#512#128
     channel = 1

phoenix_diffusion.sbatch

@@ -0,0 +1,24 @@
+#!/bin/bash
+#SBATCH -J diffusion # Job name
+#SBATCH -A gts-jw254-coda20 
+#SBATCH -qembers
+#SBATCH -N2 -c1 --gpus=RTX_6000:2 # -C A100-80GB              # Number of nodes and cores per node required
+#SBATCH --mem-per-cpu=32G                        # Memory per core
+#SBATCH -t 10:00                                    # Duration of the job (Ex: 15 mins)
+#SBATCH -oReport-%j                         # Combined output and error messages file
+#SBATCH --mail-type=BEGIN,END,FAIL              # Mail preferences
+
+#module load gcc/10.3.0-o57x6h
+#module load mvapich2/2.3.6-ouywal
+
+pwd
+date
+module load anaconda3/2022.05 # Load module dependencies
+module load pytorch
+module list
+conda activate diffusers 
+conda env list
+
+cat $0
+
+srun python diffusion.py --train 1 --sample 0