package version

[KunChenUSTC]

Hi! I want to train an autoencoder, and run the train_code_finder.py code. In your README.md, the python version should be >=3.8, while in your requirements.txt you need kornia==0.3.1. This is kind of confusing when I tried to install kornia==0.3.1 and found kornia==0.3.1 did not support python>=3.8.0. So what am I missing, or what can I do to make it work? Thank you very much!

[Bartzi]

Hi,

this is indeed very odd. Could you provide me with the exact error?

[KunChenUSTC]

Hi, This is the error that occurred when I tried to install 'kornia'. And I have a RuntimeError: Unknown type name 'torch.dtype': File "E:\anaconda\envs\one-model-to-reconstruct-them-all-main\lib\site-packages\kornia\utils\grid.py", line 12. This one occurred in trian_code_finder.py line 14, in 'from evaluation.autoencoder_evaluation import AutoEncoderEvalFunc'

[Bartzi]

I see you are using anaconda. This could be the problem because according to the Kornia release notes it should support Python 3.8. If it really does not work, you can try to use newer versions of Kornia. However, there might be problems with the SSIM calculation.

If you use the provided Dockerfile to build a docker container, it should work. If not I'll have to take a closer look but this might take some time.

[KunChenUSTC]

Thanks a lot! I will try the Dockerfile way!