add experiments

experiments/config.yaml

@@ -1,5 +1,5 @@
 # lightning.pytorch==2.5.5
-seed_everything: true
+seed_everything: 1999
 trainer:
   accelerator: gpu
   strategy: auto
@@ -14,7 +14,13 @@ trainer:
         mode: min
         save_top_k: 1
         filename: best-checkpoint
-  max_epochs: 50
+    - class_path: lightning.pytorch.callbacks.EarlyStopping
+      init_args:
+        monitor: val/loss
+        mode: min
+        patience: 10
+        verbose: false
+  max_epochs: 200
   min_epochs: null
   max_steps: -1
   min_steps: null
@@ -22,6 +28,8 @@ trainer:
   log_every_n_steps: null
   inference_mode: true
   default_root_dir: null
+  accumulate_grad_batches: 2
+  gradient_clip_val: 1.0
 model: 
   class_path: ThermalSolver.module.GraphSolver
   init_args:
@@ -29,17 +37,17 @@ model:
     model_init_args:
       x_ch_node: 1 
       f_ch_node: 1 
-      hidden: 16 
-      layers: 2
+      hidden: 16
+      layers: 1
       edge_ch: 3 
       out_ch: 1
-    unrolling_steps: 10
+    unrolling_steps: 64
 data: 
   class_path: ThermalSolver.data_module.GraphDataModule
   init_args:
     hf_repo: "SISSAmathLab/thermal-conduction"
-    split_name: "2000"
-    batch_size: 6
+    split_name: "2000_ref_1"
+    batch_size: 10
     train_size: 0.8
     test_size: 0.1
     test_size: 0.1