transfer files

experiments/config_01.yaml

@@ -9,8 +9,8 @@ trainer:
   logger:
     - class_path: lightning.pytorch.loggers.TensorBoardLogger
       init_args:
-        save_dir: lightning_logs
-        name: "01"
+        save_dir: logs
+        name: "test"
         version: null
   callbacks:
     - class_path: lightning.pytorch.callbacks.ModelCheckpoint
@@ -33,26 +33,21 @@ trainer:
   log_every_n_steps: null
   inference_mode: true
   default_root_dir: null
-  accumulate_grad_batches: 6
-  # gradient_clip_val: 1.0
+  accumulate_grad_batches: 4
+  gradient_clip_val: 1.0
 model: 
-  class_path: ThermalSolver.module.GraphSolver
+  class_path: ThermalSolver.graph_module.GraphSolver
   init_args:
-    model_class_path: ThermalSolver.model.local_gno.GatingGNO
+    model_class_path: ThermalSolver.model.LearnableGraphFiniteDifference
     model_init_args:
-      x_ch_node: 1 
-      f_ch_node: 1 
-      hidden: 16
-      layers: 1
-      edge_ch: 3 
-      out_ch: 1
+      max_iters: 250
     unrolling_steps: 64
 data: 
-  class_path: ThermalSolver.data_module.GraphDataModule
+  class_path: ThermalSolver.graph_datamodule.GraphDataModule
   init_args:
     hf_repo: "SISSAmathLab/thermal-conduction"
-    split_name: "2000_ref_1"
-    batch_size: 4
+    split_name: "1000_40x30"
+    batch_size: 8
     train_size: 0.8
     test_size: 0.1
     test_size: 0.1