| 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: | |
| name: "KNO2DNavierStokes" | |
| input_channels: 10 | |
| output_channels: 1 | |
| op_size: 32 | |
| modes_x: 12 | |
| modes_y: 12 | |
| decompose: 6 | |
| linear_type: true | |
| normalization: false | |
| training: | |
| epochs: 500 | |
| lr: 0.001 | |
| weight_decay: 0.00001 | |
| step_size: 100 | |
| gamma: 0.5 | |
| eval_interval: 10 | |
| patience: 100 | |
| teacher_forcing: true | |
| max_grad_norm: null | |
| max_train_batches: null | |
| max_eval_batches: null | |
| weight_dir: "weight" | |
| checkpoint_name: "kno_navier_stokes.pt" | |
| inference: | |
| result_dir: "result" | |
| num_samples: 4 | |