fix experiments

This commit is contained in:
Filippo Olivo
2025-12-12 10:18:49 +01:00
parent 732d48c360
commit 3cc1d230e4
3 changed files with 6 additions and 6 deletions

View File

@@ -58,7 +58,7 @@ data:
class_path: ThermalSolver.graph_datamodule_unsteady.GraphDataModule class_path: ThermalSolver.graph_datamodule_unsteady.GraphDataModule
init_args: init_args:
hf_repo: "SISSAmathLab/thermal-conduction-unsteady" hf_repo: "SISSAmathLab/thermal-conduction-unsteady"
split_name: "easy.refined" split_name: "basic.refined"
n_elements: 100 n_elements: 100
batch_size: 32 batch_size: 32
train_size: 0.7 train_size: 0.7

View File

@@ -39,7 +39,7 @@ trainer:
min_steps: null min_steps: null
overfit_batches: 0.0 overfit_batches: 0.0
log_every_n_steps: 0 log_every_n_steps: 0
accumulate_grad_batches: 2 accumulate_grad_batches: 1
default_root_dir: null default_root_dir: null
gradient_clip_val: 1.0 gradient_clip_val: 1.0
@@ -58,9 +58,9 @@ data:
class_path: ThermalSolver.graph_datamodule_unsteady.GraphDataModule class_path: ThermalSolver.graph_datamodule_unsteady.GraphDataModule
init_args: init_args:
hf_repo: "SISSAmathLab/thermal-conduction-unsteady" hf_repo: "SISSAmathLab/thermal-conduction-unsteady"
split_name: "easy.refined" split_name: "basic.refined"
n_elements: 100 n_elements: 100
batch_size: 16 batch_size: 24
train_size: 0.7 train_size: 0.7
val_size: 0.2 val_size: 0.2
test_size: 0.1 test_size: 0.1

View File

@@ -24,7 +24,7 @@ trainer:
init_args: init_args:
monitor: val/loss monitor: val/loss
mode: min mode: min
patience: 20 patience: 30
verbose: false verbose: false
max_epochs: 1000 max_epochs: 1000
min_epochs: null min_epochs: null
@@ -50,7 +50,7 @@ data:
class_path: ThermalSolver.graph_datamodule_unsteady.GraphDataModule class_path: ThermalSolver.graph_datamodule_unsteady.GraphDataModule
init_args: init_args:
hf_repo: "SISSAmathLab/thermal-conduction-unsteady" hf_repo: "SISSAmathLab/thermal-conduction-unsteady"
split_name: "easy.refined" split_name: "basic.refined"
n_elements: 100 n_elements: 100
batch_size: 32 batch_size: 32
train_size: 0.7 train_size: 0.7