fix experiments
This commit is contained in:
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
@@ -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
|
||||||
|
|||||||
Reference in New Issue
Block a user