File size: 694 Bytes
7f71cfd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
seed: 42
data:
  file: data/stablenn_phys_virtual.npz
  format_version: stablenn_phys_v1
  levels: 34
  window_steps: 20
  columns: 6
  windows_per_column: 4
  step_hours: 3
model:
  name: StableNN-Phys
  hidden_size: 32
  paper_hidden_size: 128
  input_features: 71
  output_features: 68
train:
  epochs: 3
  batch_size: 4
  learning_rate: 0.003
  loss_mode: paper
  paper_model:
    hidden_size: 128
    learning_rate: 0.01
    batch_size: 200
    epochs: 5
runtime:
  device: auto
  rollout_steps: 64
paths:
  checkpoint: result/checkpoints/stablenn_phys.pt
  training_metrics: result/training/metrics.json
  inference: result/output/rollout.npz
  evaluation: result/evaluation/metrics.json