add experiments

2025-12-18 09:30:37 +01:00
parent 0a034225ef
commit db50f5ed69
9 changed files with 397 additions and 36 deletions
--- a/experiments/10_steps/config_16_layer_16_hidden_refined.yaml
+++ b/experiments/10_steps/config_16_layer_16_hidden_refined.yaml
@@ -29,10 +29,10 @@ trainer:
    - class_path: ThermalSolver.switch_dataloader_callback.SwitchDataLoaderCallback
      init_args:
        increase_unrolling_steps_by: 4
-        patience: 15
-        last_patience: 20
+        patience: 5
+        last_patience: 15
        max_unrolling_steps: 10
-        ckpt_path: logs.autoregressive.wandb/10_steps/basic.refined.combined/16_layer_16_hidden/
+        ckpt_path: logs.autoregressive.wandb/10_steps/basic.refined/16_layer_16_hidden/
  max_epochs: 1000
  min_epochs: null
  max_steps: -1
@@ -59,13 +59,14 @@ data:
  init_args:
    hf_repo: "SISSAmathLab/thermal-conduction-unsteady"
    split_name: "3_stripes.basic.refined"
-    n_elements: 100
-    batch_size: 32
+    n_elements: 50
+    batch_size: 24
    train_size: 0.7
    val_size: 0.2
    test_size: 0.1
    build_radial_graph: false
    remove_boundary_edges: true
    unrolling_steps: 2
+    min_normalized_diff: 1e-4
 optimizer: null
 lr_scheduler: null