Hugging Face's logo

OneScience
/
BENO

English
OneScience
BENO
elliptic PDEs
boundary-embedded neural operator
Model card Files
xet
 Community
BENO
File size: 1,282 Bytes
7087ec2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
root:
  datapipe:
    source:
      data_dir: "${ONESCIENCE_BENO_DATA_DIR}/Dirichlet/"
      cache_dir: "./cache_data"
      file_prefix: "N8_smoke"
    verbose: false

    data:
      name: "BENO_Elliptic"
      ntrain: 2
      ntest: 1
      resolution: 8
      ns: 4
      domain_bounds: [[0, 1], [0, 1]]

    dataloader:
      batch_size: 1
      num_workers: 0
      pin_memory: false

  model:
    name: "HeteroGNS"
    nnode_in_features: 10
    nnode_out_features: 1
    nedge_in_features: 7
    nmlp_layers: 2
    act: "relu"
    boundary_dim: 16
    trans_layer: 1
    latent_dim: 16
    nmessage_passing_steps: 1
    mlp_hidden_dim: 16

  training:
    device: "cpu"
    output_dir: "./weight"
    epochs: 1
    save_period: 1
    checkpoint_name: "model_epoch_{epoch}.pt"
    seed: 2025

    optimizer:
      name: "Adam"
      lr: 1.0e-3
      weight_decay: 5.0e-4

    scheduler:
      name: "CosineAnnealingWarmRestarts"
      T_0: 2
      T_mult: 1

  inference:
    device: "cpu"
    checkpoint_dir: "./weight"
    checkpoint_path: ""
    result_dir: "./result/output"
    picture_dir: "./result/picture"
    metrics_path: "./result/metrics.json"
    num_visualize: 1

  fake_data:
    root_dir: "./data"
    seed: 2025
    boundary_points: 128
    noise_std: 0.001