# lightning.pytorch==2.5.5
seed_everything: true
trainer:
  accelerator: gpu
  strategy: auto
  devices: 1
  num_nodes: 1
  precision: null
  logger: null
  callbacks:
    - class_path: lightning.pytorch.callbacks.ModelCheckpoint
      init_args:
        monitor: val/loss
        mode: min
        save_top_k: 1
        filename: best-checkpoint
  max_epochs: 50
  min_epochs: null
  max_steps: -1
  min_steps: null
  overfit_batches: 0.0
  log_every_n_steps: null
  inference_mode: true
  default_root_dir: null
model: 
  class_path: ThermalSolver.module.GraphSolver
  init_args:
    model_class_path: ThermalSolver.model.local_gno.GatingGNO
    model_init_args:
      x_ch_node: 1 
      f_ch_node: 1 
      hidden: 16 
      layers: 2
      edge_ch: 3 
      out_ch: 1
    unrolling_steps: 10
data: 
  class_path: ThermalSolver.data_module.GraphDataModule
  init_args:
    hf_repo: "SISSAmathLab/thermal-conduction"
    split_name: "2000"
    batch_size: 6
    train_size: 0.8
    test_size: 0.1
    test_size: 0.1
optimizer: null
lr_scheduler: null
ckpt_path: null