search

nickyisadog / latent-diffusion-inpainting

77 stars 14 forks source link

A question about finetune the autoencoder #14

Closed RunpuWei closed 8 months ago

RunpuWei commented 8 months ago

Hello author, thank you for such a perfect job! Now I want to use your code to do inpainting of colonoscopy polyps.

I would like to ask a question of step 1 (Finetune the autoencoder), I don't seem to find the code related to stopping training in the code, is your code also not automatically stopping training? If yes, how many epoch or iteration did you finetune on the colonoscopy polyp dataset?

Looking forward to your reply!

Polaris0421 commented 4 months ago

hi, have you figured out the question? I try to find the code setting up the number of epochs but I fail

ultiwinter commented 3 weeks ago

@RunpuWei Hi, thanks for putting the question out there, would you mind sharing your solution you found?