| root: | |
| common: | |
| device: "auto" | |
| seed: 0 | |
| datapipe: | |
| source: | |
| data_dir: "/public/share/sugonhpcapp01/onestore/onedatasets/CFD_Benchmark/ns" | |
| file_name: "NavierStokes_V1e-5_N1200_T20.mat" | |
| data: | |
| ntrain: 1000 | |
| ntest: 200 | |
| t_in: 10 | |
| t_out: 10 | |
| out_dim: 1 | |
| downsamplex: 1 | |
| downsampley: 1 | |
| normalize: false | |
| dataloader: | |
| batch_size: 8 | |
| num_workers: 4 | |
| pin_memory: true | |
| model: | |
| in_dim: 10 | |
| out_dim: 1 | |
| hidden_dim: 64 | |
| modes: 12 | |
| num_layers: 4 | |
| space_dim: 2 | |
| include_pos: true | |
| activation: "gelu" | |
| reflection: false | |
| pad_to_multiple: 16 | |
| training: | |
| epochs: 500 | |
| lr: 0.001 | |
| weight_decay: 0.00001 | |
| step_size: 100 | |
| gamma: 0.5 | |
| eval_interval: 1 | |
| patience: 100 | |
| teacher_forcing: false | |
| max_grad_norm: null | |
| max_train_batches: null | |
| max_eval_batches: null | |
| weight_dir: "weight" | |
| checkpoint_name: "gfno_navier_stokes.pt" | |
| inference: | |
| result_dir: "result" | |
| num_samples: 4 | |