building and training a custom dataset

[fn6767]

Excuse me, if I want to train on my own dataset for image restoration and completion tasks, is the img size limited to 256? What size can I set? Can I continue training on your provided pre-trained models with my own dataset?

[Algolzw]

Hi! Sure, you can finetune the model on your dataset. The size is also not limited. We usually crop images with a fixed size during training but it won't affect the inference.

[fn6767]

Thanks!