DifFace / configs /sample /iddpm_ffhq512.yaml
Zongsheng
first upload
b2aaa70
raw history blame
No virus
877 Bytes
gpu_id: ""
seed: 10000
display: True
im_size: 512
diffusion:
target: models.script_util.create_gaussian_diffusion
params:
steps: 1000
learn_sigma: True
sigma_small: False
noise_schedule: linear
use_kl: False
predict_xstart: False
rescale_timesteps: False
rescale_learned_sigmas: True
timestep_respacing: "1000"
model:
target: models.unet.UNetModel
ckpt_path: pretrained_zoo/iddpm_ffhq512/ema0999_model_500000.pth
params:
image_size: 512
in_channels: 3
model_channels: 32
out_channels: 6
attention_resolutions: [32, 16, 8]
dropout: 0
channel_mult: [1, 2, 4, 8, 8, 16, 16]
num_res_blocks: [1, 2, 2, 2, 2, 3, 4]
conv_resample: True
dims: 2
use_fp16: False
num_head_channels: 64
use_scale_shift_norm: True
resblock_updown: False
use_new_attention_order: False
model_ir: ~