File size: 724 Bytes
ef2ae28 | 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 | seed: 42
data:
root: data
format_version: wildfire_danger_spatiotemporal_v1
train_samples: 12
test_samples: 8
sequence_days: 10
channels: 25
patch_height: 25
patch_width: 25
resolution_km: 1
input_layout: BTCHW
target_layout: B1
model:
input_channels: 25
hidden_channels: 4
kernel_size: 3
dropout: 0.1
train:
epochs: 2
batch_size: 4
learning_rate: 0.003
optimizer: Adam
loss: BCEWithLogitsLoss
weight_decay: 0.0001
gradient_clip_norm: 5.0
num_workers: 0
runtime:
device: auto
paths:
checkpoint: result/checkpoints/firecubenet.pt
training_metrics: result/training/metrics.json
inference_dir: result/output
evaluation_dir: result/evaluation
evaluation:
threshold: 0.5
|