search

ChenWu98 / cycle-diffusion

[ICCV 2023] A latent space for stochastic diffusion models
Other
562 stars 35 forks source link

How to write config.yaml for custom datasets #31

Closed loliq closed 5 months ago

loliq commented 6 months ago

Thanks for sharing your work.

I want to train my customed unpaired image to image tasks. And I found this tutorial . But I have a question about how to write customed yaml like afhq.yaml:

data:
    dataset: "AFHQ"
    category: "dog"
    image_size: 256
    channels: 3
    logit_transform: false
    uniform_dequantization: false
    gaussian_dequantization: false
    random_flip: true
    rescaled: true
    num_workers: 0

diffusion:
    beta_schedule: linear
    beta_start: 0.0001
    beta_end: 0.02
    num_diffusion_timesteps: 1000

Is it automatically generated or can I customize it?

Looking for your replying. Thanks.

ChenWu98 commented 6 months ago

Hi! Sorry but can you provide a pointer to the file you referred to?