ykdai / BasicPBC

Official Implementation of "Learning Inclusion Matching for Animation Paint Bucket Colorization"
Other
259 stars 23 forks source link

ModuleNotFoundError: No module named 'torch_scatte #27

Closed HideyukiOgura closed 3 months ago

HideyukiOgura commented 3 months ago

After executing pip install -r requirements.txt and python setup.py develop, I executed python inference_line_frames.py --path dataset/test/laughing_girl and ModuleNotFoundError: No module named 'torch_scatter' occurred. Please let me know how to handle this.

ykdai commented 3 months ago

You may need to check your torch version at first, and then run pip install torch-scatter following this 'https://pypi.org/project/torch-scatter/'

HideyukiOgura commented 3 months ago

Thanks for the reply. The error has been resolved.