root:
  common:
    device: "auto"
    seed: 1234

  datapipe:
    source:
      data_dir: "/public/share/sugonhpcapp01/onestore/onedatasets/Kolmogorov_flow_2d"
      file_name: "kf_2d_re1000_256_120seed.npy"
    data:
      train_num: 100
      test_num: 20
      resolution: 128
      interval: 2
      t_in: 10
      t_out: 16
      out_dim: 1
      normalize: true
      val_mode: false
      split_seed: 1234
      stats_file: "data/km2d_stat.npz"
      stats_max_trajectories: 8
    dataloader:
      batch_size: 4
      num_workers: 2
      pin_memory: true

  model:
    in_dim: 10
    out_dim: 1
    hidden_dim: 128
    depth: 4
    heads: 8
    mlp_ratio: 2
    dropout: 0.0
    activation: "gelu"
    include_pos: true
    space_dim: 2
    latent_multiplier: 2.0
    max_latent_steps: 4

  training:
    epochs: 50
    lr: 0.0005
    weight_decay: 0.0001
    step_size: 20
    gamma: 0.5
    eval_interval: 1
    patience: 20
    train_latent_steps: 4
    max_grad_norm: 1.0
    max_train_batches: null
    max_eval_batches: null
    weight_dir: "weight"
    checkpoint_name: "factformer_kolmogorov.pt"

  inference:
    result_dir: "result"
    num_samples: 4
    figure_name: "prediction_sample.png"
    tensor_name: "prediction_sample.pt"