ClimateNet / conf /config.yaml
zhangrenchao's picture
Publish ClimateNet reproduction
63ef347 verified
Raw
History Blame Contribute Delete
637 Bytes
seed: 42
data: {path: data/climatenet.npz, samples: 24, channels: 4, tile: [96, 96], logical_grid: [1152, 768], classes: 3}
model: {channels: 4, classes: 3, hidden: 16}
train: {epochs: 2, learning_rate: 0.0015, class_weights: [1, 16, 16]}
paths: {checkpoint: result/checkpoints/climatenet.pt, training_metrics: result/training/metrics.json, predictions: result/output/predictions.npz, evaluation: result/evaluation/metrics.json, figure: result/evaluation/comparison.png}
paper_model: {input_shape: [4, 1152, 768], classes: [BG, TC, AR], optimizer: Adam, learning_rate: 0.0015, batch_size: 16, expert_epochs: 5, paper_license: CC-BY-4.0}