StableNN-Phys / conf /config.yaml
zhangrenchao's picture
Upload folder using huggingface_hub
7f71cfd verified
Raw
History Blame Contribute Delete
694 Bytes
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