File size: 1,539 Bytes
338c3e4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
root:
  datapipe:
    source:
      data_dir: "./data/cfdbench"
      data_name: "tube_prop_bc_geo"
    verbose: true

    data:
      task_type: "auto"
      norm_props: true
      norm_bc: true
      delta_time: 0.1
      stable_state_diff: 0.001
      split_ratios: [0.7, 0.15, 0.15]
      seed: 0
      num_rows: 8
      num_cols: 8

    dataloader:
      batch_size: 2
      eval_batch_size: 2
      num_workers: 0

  model:
    name: "fno"
    num_label_samples: 64
    in_chan: 2
    out_chan: 2
    ffn_depth: 2
    ffn_width: 64
    autoffn_depth: 2
    autoffn_width: 64
    deeponet_width: 64
    branch_depth: 2
    trunk_depth: 2
    act_fn: "relu"
    act_scale_invariant: true
    act_on_output: false
    autoedeeponet_width: 64
    autoedeeponet_depth: 2
    autoedeeponet_act_fn: "relu"
    fno_depth: 2
    fno_hidden_dim: 16
    fno_modes_x: 4
    fno_modes_y: 4
    unet_dim: 8
    unet_insert_case_params_at: "input"
    resnet_depth: 2
    resnet_hidden_chan: 16
    resnet_kernel_size: 7
    resnet_padding: 3

  training:
    output_dir: "./result"
    mode: "train_test"
    group_by_model: true
    device: "cpu"
    num_epochs: 1
    lr: 1.0e-3
    lr_step_size: 20
    lr_gamma: 0.9
    log_interval: 1
    eval_interval: 1
    loss_name: "nmse"
    checkpoint_name: "auto"

  inference:
    checkpoint_path: "auto"
    output_dir: "./result/inference"
    group_by_model: true
    device: "cpu"
    max_batches: 2
    plot_interval: 1

  fake_data:
    num_cases_per_subset: 4
    num_frames: 4
    seed: 123