Colab Implementation - Githubissues

shazib-summar commented 4 years ago

Hi @shahidul56, Thanks for your implementation of ESRGAN in Colab. I'm facing an issue, when I run the notebook in colab I face a few errors which for the life of me I cannot resolve. I've attached them below. Any help would be appreciated. Thanks a lot.

errors_colab_esrgan

shazib-summar commented 4 years ago

Nevermind, figured it out.

CloudEPDB commented 4 years ago

I have the same problem, always show me this problem Sin título

shazib-summar commented 4 years ago

To solve the issue, make the following modification to the original code.

The link used in the wget command is for the old architecture. This should work with older version of PyTorch but it does not, so update the command with this one !wget --no-check-certificate 'https://docs.google.com/uc?export=download&id=1TPrz5QKd8DHHt1k8SRtm6tMiPjz_Qene' -O models/RRDB_ESRGAN_x4.pth
Also, the version of PyTorch is deprecated so replace the command !pip install -q http://download.pytorch.org/whl/{accelerator}/torch-0.4.1-{platform}-linux_x86_64.whl with !pip install torchvision to install the latest version.

The latest model architecture would work on the latest version of PyTorch.

You're welcome.

CloudEPDB commented 4 years ago

Thanks so much killerzula!

But, then i can't use pth with "old arch" like DigitalFrames_2.1? I need to use this pth. Now, only can use RRDB_ESRGAN_x4.pth

Thanks for your great support

styler00dollar commented 4 years ago

@CloudEPDB I edited this notebook and improved it in several ways. It does also use old-arch and should be compatible with everything important. You can find my Colab-ESRGAN fork with that notebook here.

xinntao / ESRGAN

Colab Implementation #22