# https://github.com/neuraloperator/physics_informed/blob/master/configs/scratch/Re300-scratch-1s.yaml
data:
  # datapath: 'data/NS_fine_Re300_T128.npy'
  paths: ['../data/NS_fine_Re300_T128_part0.npy', '../data/NS_fine_Re300_T128_part1.npy', '../data/NS_fine_Re300_T128_part2.npy']
  Re: 300
  total_num: 300
  offset: 0 # 250
  raw_res: [128, 128, 129]
  n_data_samples: 250
  data_res: [64, 64, 129]  # resolution in 1 second
  # data_res: [64, 64, 65]  # resolution in 1 second
  pde_res: [64, 64, 129]   # resolution in 1 second
  a_offset: 0
  n_a_samples: 250
  testoffset: 250
  n_test_samples: 50
  t_duration: 1
  shuffle: True
  mask_ratio: None # 0~1 or None


model:
  # layers: [64, 64, 64, 64, 64]
  # fc_dim: 128
  layers: [128, 128, 128, 128, 128]
  fc_dim: 256
  modes1: [12, 12, 12, 12]
  modes2: [12, 12, 12, 12]
  modes3: [12, 12, 12, 12]
  act: gelu
  pad_ratio: [0, 1]

train:
  batchsize: 2
  start_iter: 0
  num_iter: 50_001
  milestones: [20_000, 40_000]
  # num_iter: 8001
  # milestones: [1000, 2000, 3000, 4000, 5000, 6000, 7000]
  # base_lr: 0.001
  base_lr: 0.0025
  scheduler_gamma: 0.5
  ic_loss: 0.0
  f_loss: 0.0
  xy_loss: 1.0
  save_step: 10000
  eval_step: 5

test:
  batchsize: 1
  data_res: [64, 64, 129]
  # data_res: [64, 64, 65]
  ckpt: model-400.pt

log:
  logdir: Re300-1s-FNO-s
  entity: pinns
  project: foundationmodels
  group: Re300-1s-FNO-s
