CRAI-ClimateExtremes / conf /config.yaml
zhangrenchao's picture
Publish CRAI-ClimateExtremes reproduction
b20ca9c verified
Raw History Blame Contribute Delete
392 Bytes
seed: 42
data_path: data/crai_fake.npz
height: 144
width: 192
base_channels: 8
batch_size: 2
epochs: 1
max_iterations: 8
learning_rate: 0.00005
ensemble_members: 2
paper_batch_size: 16
paper_iterations: 1000000
paper_ensemble_members: 20
paper_model: false
num_samples: 8
checkpoint_path: result/checkpoints/crai_climateextremes.pt
output_dir: result/output
evaluation_dir: result/evaluation