[Colab] cannot import name 'filter2d' from 'kornia'

[katongo]

Hi

Training on google Colab Pro, I started getting this error:

Traceback (most recent call last):
File "/usr/local/bin/lightweight_gan", line 5, in
from lightweight_gan.cli import main
File "/usr/local/lib/python3.7/dist-packages/lightweight_gan/init.py", line 1, in
from lightweight_gan.lightweight_gan import LightweightGAN, Generator, Discriminator, Trainer, NanException
File "/usr/local/lib/python3.7/dist-packages/lightweight_gan/lightweight_gan.py", line 25, in
from kornia import filter2d
ImportError: cannot import name 'filter2d' from 'kornia' (/usr/local/lib/python3.7/dist-packages/kornia/init.py)

Before that everything worked fine for several weeks

[arturocalvoacn]

It seems to be connected with the release 0.6.1 on Kornia (Oct 22nd), where they refactored the library namespaces.
kornia/kornia@218232c

I think filter2d is now under kornia.filters instead of directly under kornia. This causes the training and generation functions of lightweightGAN to crash.

In the meantime you can continue training / generating by reverting to the previous release on Kornia:
pip install kornia==0.5.4

FYI @lucidrains

[katongo]

Thank you @arturocalvoacn. Indeed downgrading to kornia 0.5.4 fixes the issue.
I guess a fix in the project imports or dependencies is still required

[CaptainStabs]

This was fixed with the latest release

[domef]

I'm having the same problem, I fixed it by changing

lightweight-gan/lightweight_gan/lightweight_gan.py

Line 25 in fce2093

from kornia import filter2d

in from kornia.filters import filter2d.