[unet] batch_size = 256 time_emb_dim = 128 device = cuda epoch_num = 500 learning_rate = 1e-4 [ddpm] iteration = 500