search

Fantasy-Studio / Paint-by-Example

Paint by Example: Exemplar-based Image Editing with Diffusion Models
https://arxiv.org/abs/2211.13227
Other
1.03k stars 93 forks source link

Which model we should select as the final inference model among multiple saved models? #20

Open Delicious-Bitter-Melon opened 1 year ago

Delicious-Bitter-Melon commented 1 year ago

Thanks for your excellent work.

The training code saves multiple models during training. Which model we should select as the final inference model?

Delicious-Bitter-Melon commented 1 year ago

It seems that a model with minimal "val/loss_simple_ema" is selected.

Delicious-Bitter-Melon commented 1 year ago

can you guid me with the full steps u made to train your model

You need to modify this file "https://github.com/Fantasy-Studio/Paint-by-Example/blob/main/ldm/data/open-images.py" for your custom dataset.