TropiCycloneNet / conf /config.yaml
zhangrenchao's picture
Publish TropiCycloneNet reproduction
01384b4 verified
Raw History Blame Contribute Delete
785 Bytes
seed: 42
data:
format_version: tcnd_synthetic_v1
manifest: data/manifest.json
samples: 8
history_steps: 8
forecast_steps: 4
interval_hours: 6
data1d_features: 4
data3d_channels: 1
grid: [81, 81]
environment_features: 73
basins: [NA, EP, WP, NI, SI, SP]
model:
hidden_dim: 24
generators: 6
heads: 4
train:
epochs: 1
batch_size: 2
learning_rate: 0.001
paths:
checkpoint: result/checkpoints/tropicyclonenet.pt
training_metrics: result/training/metrics.json
predictions: result/output/predictions.npz
evaluation: result/evaluation/metrics.json
figure: result/evaluation/comparison.png
paper_model:
training_years: [1950, 2016]
test_years: [2017, 2021]
optimizer: Adam
learning_rate: 0.0001
batch_size: 96
epochs: 102
generators: 6