search

Janspiry / Image-Super-Resolution-via-Iterative-Refinement

Unofficial implementation of Image Super-Resolution via Iterative Refinement by Pytorch
Apache License 2.0
3.6k stars 469 forks source link

Question of pretrained model #138

Open yxldd opened 11 months ago

yxldd commented 11 months ago

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

include5636 commented 10 months ago

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