how can i train my own data ,can you provide a detailed description?

andreas128 / RePaint

Official PyTorch Code and Models of "RePaint: Inpainting using Denoising Diffusion Probabilistic Models", CVPR 2022

1.88k stars 153 forks source link

Open No360201 opened 1 year ago

No360201 commented 1 year ago

i find the unet in https://github.com/openai/improved-diffusion is different from yours in the project ,can you help me? @andreas128 @freddy1020

andreas128 commented 1 year ago

To get the same commit as we used do the following:

git clone https://github.com/openai/guided-diffusion.git
cd guided-diffusion
git checkout 912d577

For training we use 256 × 256 crops in three batches on 4×V100 GPUs each.

Does that work for you?

xyz-xdx commented 1 year ago

@No360201 Can you use your own dataset to train the model?