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