-
Notifications
You must be signed in to change notification settings - Fork 0
/
train.yml
52 lines (39 loc) · 843 Bytes
/
train.yml
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
42
43
44
45
46
47
48
49
50
51
52
### conf/train.yml
#
# Train Hyperparameters
#
run_name : DDPM-Pyramid
epochs : 233
batch_size : 128
dataset_path : "./data/BraTS21/"
workers : 4 # cpu data loaders
start_lr : 2.0e-5
target_lr : 1.0e-4
warmup_steps: 0.05
ema_decay: 0.995
horizontal_flip: # leave empty if not needed
# The checkpoints need to be given when continuing the training
train_continue : False
model :
ema_model :
opt :
start_ckpt : 120 # first checkpoint that is safed, integer equals number of epochs
save_ckpt : 4 # number of epochs between each checkpoint
num_images : 3 # number of sampled images during training
#
# Model Hyperparameters
#
size : 128
channels : 4
#
# Diffusion Hyperparameters
#
noise_steps: 1000
beta_start: 1.0e-4
beta_end: 0.02
#
# Noise type
#
pyramid : True
simplex : False
discount : 0.8 #only needed for pyramid noise