10 lines
118 B
INI
10 lines
118 B
INI
[unet]
|
|
batch_size = 256
|
|
time_emb_dim = 128
|
|
device = cuda
|
|
epoch_num = 500
|
|
learning_rate = 1e-4
|
|
|
|
[ddpm]
|
|
iteration = 500
|