av-savchenko / hsemotion

HSEmotion Python library for facial expression recognition
Apache License 2.0
26 stars 7 forks source link

return super().find_class(mod_name, name) ModuleNotFoundError: No module named 'timm.layers' #5

Closed monajalal closed 2 months ago

monajalal commented 2 months ago

hsemotion/facial_emotions.py", line 50, in __init__
    model=torch.load(path, map_location=torch.device('cpu'))
  File "/home/mona/.local/lib/python3.10/site-packages/torch/serialization.py", line 1097, in load
    return _load(
  File "/home/mona/.local/lib/python3.10/site-packages/torch/serialization.py", line 1525, in _load
    result = unpickler.load()
  File "/home/mona/.local/lib/python3.10/site-packages/torch/serialization.py", line 1515, in find_class
    return super().find_class(mod_name, name)
ModuleNotFoundError: No module named 'timm.layers'

I installed using setup.py

Could you please help in fixing this?

monajalal commented 2 months ago

fixed w pip install timm==0.9.7