search

Zhendong-Wang / Prompt-Diffusion

Official PyTorch implementation of the paper "In-Context Learning Unlocked for Diffusion Models"
Apache License 2.0
364 stars 10 forks source link

training epoch #7

Open Jasonhyw opened 9 months ago

Jasonhyw commented 9 months ago

Hi, How many epoch did your checkpoint train ？Thanks.

Zhendong-Wang commented 8 months ago

We finetune Prompt Diffusion for 5000 steps, nearly 20 epochs, with batch size 256*4=1024.

Jasonhyw commented 8 months ago

We finetune Prompt Diffusion for 5000 steps, nearly 20 epochs, with batch size 256*4=1024.

I trained it for 9999 steps, nearly 100 epochs, with batch size 172*4=688. But when I valued it, the resulting image is a pink noise image.I don't know the reason. May you have any advice? Thanks.

Zhendong-Wang commented 8 months ago

The reason could be the datasets. I am not sure what dataset you use. I got confirmed from others the code could reproduce the original results with Instruct Pix2Pix modified data.