Hugging Face's logo

Xsmos
/
ml21cm

TensorBoard
generate 21cm lightcones
denoising diffusion probabilistic model
Model card Files
xet
Metrics Community
Xsmos commited on
Commit
c6c1211
·
verified ·
1 Parent(s): 066bcd7

21012525

Browse files
Files changed (6) hide show
  1. .gitattributes +1 -0
  2. 30752459 +0 -0
  3. 30752582 +0 -0
  4. 30752793 +0 -0
  5. perlmutter_diffusion.sbatch +7 -7
  6. quantify_results.ipynb +0 -0
.gitattributes CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ quantify_results.ipynb filter=lfs diff=lfs merge=lfs -text
30752459 ADDED
The diff for this file is too large to render. See raw diff
 
30752582 ADDED
The diff for this file is too large to render. See raw diff
 
30752793 ADDED
The diff for this file is too large to render. See raw diff
 
perlmutter_diffusion.sbatch CHANGED
@@ -5,9 +5,9 @@
5
  #SBATCH -q shared #regular
6
  #SBATCH -N1
7
  #SBATCH --gpus-per-node=1
8
- #SBATCH -t 3:00:00
9
  #SBATCH --ntasks-per-node=1
10
- #SBATCH -o%j
11
  #SBATCH --mail-type=BEGIN,END,FAIL
12
  #SBATCH --gpu-bind=none
13
 
@@ -36,16 +36,16 @@ cat $0
36
  #nvidia-smi
37
 
38
  srun python diffusion.py \
39
- --num_image 6400 \
40
  --batch_size 64 \
41
  --n_epoch 50 \
42
  --gradient_accumulation_steps 1 \
43
- --num_new_img_per_gpu 200 \
44
- --max_num_img_per_gpu 20 \
45
- --channel_mult 1 1 2 2 4 \
46
  --autocast 1 \
47
  --use_checkpoint 1 \
48
  --train "$SCRATCH/LEN128-DIM64-CUB16-Tvir[4, 6]-zeta[10, 250]-0809-123640.h5" \
49
- #--resume outputs/model-N3200-device_count1-node1-epoch99-17160118 \
50
 
51
  date
 
5
  #SBATCH -q shared #regular
6
  #SBATCH -N1
7
  #SBATCH --gpus-per-node=1
8
+ #SBATCH -t 6:00:00
9
  #SBATCH --ntasks-per-node=1
10
+ #SBATCH -oReport-%j
11
  #SBATCH --mail-type=BEGIN,END,FAIL
12
  #SBATCH --gpu-bind=none
13
 
 
36
  #nvidia-smi
37
 
38
  srun python diffusion.py \
39
+ --num_image 12800 \
40
  --batch_size 64 \
41
  --n_epoch 50 \
42
  --gradient_accumulation_steps 1 \
43
+ --num_new_img_per_gpu 60 \
44
+ --max_num_img_per_gpu 60 \
45
+ --channel_mult 1 1 2 2 4 4 \
46
  --autocast 1 \
47
  --use_checkpoint 1 \
48
  --train "$SCRATCH/LEN128-DIM64-CUB16-Tvir[4, 6]-zeta[10, 250]-0809-123640.h5" \
49
+ #--resume outputs/model-N6400-device_count1-node1-epoch49-20153915 \
50
 
51
  date
quantify_results.ipynb CHANGED
The diff for this file is too large to render. See raw diff