Error while running simple_extractor.py

[soumyajit123dev]

I ran the following command

python simple_extractor.py --dataset 'lip' --input-dir 'images' --model-restore 'pretrained/exp-schp-201908261155-lip.pth' --output-dir 'output'

input image is kept inside images folder
pretrained model is kept inside pretrained folder

I am getting some errors. I do not know how to resolve these errors. Please see the errors below

:

Let me know what is required to solve the errors

[jutanke]

@soumyajit123dev resetting pytorch to an older version solved the issue on my end:
conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.2 -c pytorch

[GoGoDuck912]

Please try to run the CoLab notebook following https://colab.research.google.com/drive/1JOwOPaChoc9GzyBi5FUEYTSaP2qxJl10?usp=sharing.
If this works for you, basically the issue is related to the environment configuration based on your circumstances.