folivo/thermal-conduction-ml
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

transfer files

Browse Source
This commit is contained in:
Filippo Olivo
2025-11-25 19:19:31 +01:00
parent edba700d2a
commit 88bc5c05e4
13 changed files with 926 additions and 163 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
experiments/config_64.yaml
View File

@@ -33,11 +33,11 @@ trainer:
log_every_n_steps: null
inference_mode: true
default_root_dir: null
accumulate_grad_batches: 6
accumulate_grad_batches: 2
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_init_args:
@@ -49,11 +49,11 @@ model:
out_ch: 1
unrolling_steps: 1
data:
class_path: ThermalSolver.data_module.GraphDataModule
class_path: ThermalSolver.graph_datamodule.GraphDataModule
init_args:
hf_repo: "SISSAmathLab/thermal-conduction"
split_name: "2000"
batch_size: 4
split_name: "2000_ref_1"
batch_size: 10
train_size: 0.8
test_size: 0.1
test_size: 0.1