File size: 863 Bytes
90cbf31
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
38
39
40
41
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