joint-diffusion-celeba / config.yaml
tristan-deep's picture
Upload 4 files
ffd557d verified
activation: relu
batch_size: 16
beta_max: 50.0
beta_min: 0.1
channels: 64
color_mode: rgb
corrector: none
d_apply_batchnorm: false
d_channels:
- 64
- 128
- 256
- 256
d_drop_prob: 0.3
d_kernel_size: 5
d_lr: 0.0001
d_output_activation: null
d_steps: 3
dataset_name: celeba
drop_prob: null
embed_dim: 256
epochs: 100
eval_freq: 1
g_apply_batchnorm: true
g_channels:
- 256
- 256
- 256
- 128
g_drop_prob: null
g_first_dense_size: 128
g_kernel_size: 5
g_lr: 0.0001
g_output_activation: tanh
g_upmode: upconv
gp_weight: 1
image_range:
- 0
- 1
image_shape:
- 64
- 64
- 3
image_size: 64
kernel_size: 3
label_sigma: 0.05
latent_dim: 64
likelihood_weighting: false
lr: 0.0002
model_name: score
n_steps_each: 1
noise_removal: true
normalization: batch
num_plot_img: 16
num_scales: 1000
patch_size: null
predictor: euler_maruyama
probability_flow: false
reduce_mean: false
sampling_method: pc
save_freq: 10
score_backbone: NCSNv2
sde: simple
seed: 1234
sigma: 25.0
sigma_max: 90.0
sigma_min: 0.01
snr: 0.17
upmode: upconv