File size: 1,019 Bytes
a62fa9c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | 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
space_dim: 2
include_pos: true
normtype: "in"
bilinear: true
activation: "gelu"
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: "uno_navier_stokes.pt"
inference:
result_dir: "result"
num_samples: 4
|