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

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