Traceback (most recent call last):
File "main.py", line 112, in <module>
tl.set_backend('pytorch')
File "/opt/conda/lib/python3.8/site-packages/tensorly/backend/__init__.py", line 213, in set_backend
backend = cls.load_backend(backend)
File "/opt/conda/lib/python3.8/site-packages/tensorly/backend/__init__.py", line 191, in load_backend
importlib.import_module('tensorly.backend.{0}_backend'.format(backend_name))
File "/opt/conda/lib/python3.8/importlib/__init__.py", line 127, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
File "<frozen importlib._bootstrap>", line 1014, in _gcd_import
File "<frozen importlib._bootstrap>", line 991, in _find_and_load
File "<frozen importlib._bootstrap>", line 975, in _find_and_load_unlocked
File "<frozen importlib._bootstrap>", line 671, in _load_unlocked
File "<frozen importlib._bootstrap_external>", line 783, in exec_module
File "<frozen importlib._bootstrap>", line 219, in _call_with_frames_removed
File "/opt/conda/lib/python3.8/site-packages/tensorly/backend/pytorch_backend.py", line 230, in <module>
PyTorchBackend.register_method(name, getattr(torch, name))
AttributeError: module 'torch' has no attribute 'count_nonzero'
This error occurs because the pytorch 1.6.0 version does not provide count_nonzero.
This is because it is a function that starts from 1.7.0 version.
So I suggest that you have requirements.txt in your repository.
This error occurs because the pytorch 1.6.0 version does not provide count_nonzero. This is because it is a function that starts from 1.7.0 version. So I suggest that you have requirements.txt in your repository.