A question about finetune the autoencoder

[RunpuWei]

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]

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

[ultiwinter]

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