correcting extension to .fits

HITS-AIN · Oct 31, 2023 · 91c973c · 91c973c
1 parent 8922674
commit 91c973c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/experiments/illustris-power.yaml b/experiments/illustris-power.yaml
@@ -6,8 +6,8 @@ model:
     h_dim: 256
     z_dim: 3
     image_size: 363
-    rotations: 36
-    beta: 1.0e-3
+    rotations: 72
+    beta: 5.0e-4
 
 data:
   class_path: data.IllustrisSdssDataModule
@@ -18,7 +18,7 @@ data:
                        "/local_data/AIN/SKIRT_synthetic_images/TNG50/sdss/snapnum_095/data/",
                        "/local_data/AIN/SKIRT_synthetic_images/Illustris/sdss/snapnum_135/data/",
                        "/local_data/AIN/SKIRT_synthetic_images/Illustris/sdss/snapnum_131/data/"]
-    extension: fits
+    extension: .fits
     minsize: 100
     batch_size: 128
     shuffle: True
@@ -43,7 +43,7 @@ lr_scheduler:
 trainer:
   max_epochs: -1
   accelerator: gpu
-  devices: [1]
+  devices: [3]
   precision: 32
   # overfit_batches: 1
   logger:

diff --git a/experiments/illustris.yaml b/experiments/illustris.yaml
@@ -15,7 +15,7 @@ data:
                        "/local_data/AIN/SKIRT_synthetic_images/TNG50/sdss/snapnum_095/data/",
                        "/local_data/AIN/SKIRT_synthetic_images/Illustris/sdss/snapnum_135/data/",
                        "/local_data/AIN/SKIRT_synthetic_images/Illustris/sdss/snapnum_131/data/"]
-    extension: fits
+    extension: .fits
     minsize: 100
     batch_size: 128
     shuffle: True