Merge pull request #47 from i4Ds/feature/additional-configs

adds additional configs

config/ce-vae/run-1.yaml

@@ -27,6 +27,6 @@ data:
       value: 1h
 predict:
   mode:
-    value: sample
+    value: pixel
   pred_dir:
     value: "./output/predictions"

config/ce-vae/run-fhnw-2-use-geco.yaml

@@ -44,3 +44,7 @@ train:
     value: false
   use_geco:
     value: true
+devices:
+  gpus:
+    value:
+      - 2

config/ce-vae/run-fhnw-full-1-limb.yaml

@@ -0,0 +1,64 @@
+model:
+  load_path:
+    value: null
+data:
+  data_dir:
+    value: /mnt/nas05/astrodata01/astroml_data/sdomlv2_full/sdomlv2.zarr
+  dataset:
+    value: SDOMLDatasetV2
+  num_data_loader_workers:
+    value: 16
+  prefetch_factor:
+    value: 4
+  batch_size:
+    value: 32
+  sdo_ml_v2:
+    train_year:
+      value:
+        - "2010"
+        - "2012"
+        - "2013"
+        - "2015"
+        - "2017"
+        - "2018"
+        - "2020"
+    train_start_date:
+      value: null
+    train_end_date:
+      value: null
+    test_year:
+      value:
+        - "2011"
+        - "2014"
+        - "2016"
+        - "2019"
+    test_start_date:
+      value: null
+    test_end_date:
+      value: null
+    train_val_split_ratio:
+      value: 0.9
+    train_val_split_temporal_chunk_size:
+      value: 14d
+    sampling_strategy:
+      value: "chunk"
+    mask_limb:
+      value: true
+predict:
+  mode:
+    value: sample
+  pred_dir:
+    value: "./output/predictions"
+log_dir:
+  value: "./output/train-sdo-ml"
+train:
+  n_epochs:
+    value: 20
+  early_stopping:
+    value: true
+  use_geco:
+    value: true
+devices:
+  gpus:
+    value:
+      - 2

config/ce-vae/run-fhnw-full-1.yaml

@@ -7,11 +7,11 @@ data:
   dataset:
     value: SDOMLDatasetV2
   num_data_loader_workers:
-    value: 8
+    value: 16
   prefetch_factor:
-    value: 2
+    value: 4
   batch_size:
-    value: 64
+    value: 32
   sdo_ml_v2:
     train_year:
       value:
@@ -41,7 +41,7 @@ data:
     train_val_split_temporal_chunk_size:
       value: 14d
     sampling_strategy:
-      value: "default"
+      value: "chunk"
 predict:
   mode:
     value: sample

config/ce-vae/run-fhnw-full-2-256.yaml

@@ -0,0 +1,64 @@
+model:
+  load_path:
+    value: null
+  z_dim:
+    value: 256
+data:
+  data_dir:
+    value: /mnt/nas05/astrodata01/astroml_data/sdomlv2_full/sdomlv2.zarr
+  dataset:
+    value: SDOMLDatasetV2
+  num_data_loader_workers:
+    value: 16
+  prefetch_factor:
+    value: 4
+  batch_size:
+    value: 32
+  sdo_ml_v2:
+    train_year:
+      value:
+        - "2010"
+        - "2012"
+        - "2013"
+        - "2015"
+        - "2017"
+        - "2018"
+        - "2020"
+    train_start_date:
+      value: null
+    train_end_date:
+      value: null
+    test_year:
+      value:
+        - "2011"
+        - "2014"
+        - "2016"
+        - "2019"
+    test_start_date:
+      value: null
+    test_end_date:
+      value: null
+    train_val_split_ratio:
+      value: 0.9
+    train_val_split_temporal_chunk_size:
+      value: 14d
+    sampling_strategy:
+      value: "chunk"
+predict:
+  mode:
+    value: sample
+  pred_dir:
+    value: "./output/predictions"
+log_dir:
+  value: "./output/train-sdo-ml"
+train:
+  n_epochs:
+    value: 20
+  early_stopping:
+    value: true
+  use_geco:
+    value: true
+devices:
+  gpus:
+    value:
+      - 2