RePaint icon indicating copy to clipboard operation
RePaint copied to clipboard

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

Open No360201 opened this issue 3 years ago • 2 comments
trafficstars

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

No360201 avatar Sep 06 '22 07:09 No360201

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?

andreas128 avatar Nov 05 '22 03:11 andreas128

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

xyz-xdx avatar Mar 30 '23 04:03 xyz-xdx