Image-Super-Resolution-via-Iterative-Refinement icon indicating copy to clipboard operation
Image-Super-Resolution-via-Iterative-Refinement copied to clipboard

Published 20 hours ago • verified publisher icon Janspiry

Question of pretrained model

Open yxldd opened this issue 1 year ago • 1 comments

What's the different between the _opt and _gen pretrained models?

yxldd avatar Nov 26 '23 11:11 yxldd

Hi, @yxldd _opt is like .state file which contains training parameters like epoch / iter / scheduler / optimizer etc. while _gen is like .pth file which contains model parameters. You could find corresponding code here: https://github.com/Janspiry/Image-Super-Resolution-via-Iterative-Refinement/blob/01d27a7cbfa8502be1d8dbd4ee02fcbd5e44389d/model/model.py#L129C1-L141C40

include5636 avatar Jan 01 '24 18:01 include5636