ScaleAdaptiveCM / conf /config.yaml
zhangrenchao's picture
Publish ScaleAdaptiveCM reproduction
90cbf31 verified
Raw History Blame Contribute Delete
863 Bytes
seed: 42
data:
path: data/scale_adaptive_cm_fake.npz
format_version: scale_adaptive_cm_v1
samples: 4
low_grid: [60, 96]
high_grid: [240, 384]
scale_factor: 4
log_epsilon: 0.0001
model:
channels: [4, 8, 16]
time_dim: 16
sigma_data: 0.5
train:
epochs: 1
batch_size: 1
learning_rate: 0.0002
ema_decay: 0.9
runtime:
device: cpu
paths:
checkpoint: result/checkpoints/scale_adaptive_cm.pt
training_metrics: result/training/metrics.json
predictions: result/output/predictions.npz
evaluation: result/evaluation/metrics.json
figure: result/evaluation/comparison.png
evaluation:
ensemble_members: 3
guidance_sigma: 0.468
paper_model:
high_grid: [240, 384]
low_grid: [60, 96]
channels: [128, 128, 256, 256]
epochs: 150
batch_size: 1
learning_rate: 0.0002
t_min: 0.002
t_max: 80.0
rho: 7
ema_initial_decay: 0.9