model:
  name: "ViT"
  in_channels: 3
  out_channels: 3
  height: 192
  width: 384
  encoder_depth: 6
  encoder_embed_dim: 768
  encoder_num_heads: 12
  decoder_depth: 6
  decoder_embed_dim: 512
  decoder_num_heads: 16
  ckpt_path: "path/to/ckpt"

data:
  root_dir: "./dataset"
  train_file_name: "train/train_dataset.mind"
  test_file_name: "test/test_dataset.mind"
  finetune_file_name: "finetune_dataset.mind"
  grid_file_name: "grid.npy"
  finetune_ratio: 0.2
  batch_size: 32
  max_value_list: [ 4.6, 1.0, 0.18418664581293126]
  min_value_list: [-2.0, -0.18382872836312403, -0.1839052340212462]

optimizer:
  epochs: 1000
  learning_rate: 0.0005
  wave_level: 1

summary:
  summary_dir: "./summary_dir"
  test_interval: 10
  plot_interval: 100
  save_ckpt_interval: 100
  postprocess_dir: "./postprocess_dir/visualization/"