在配置forecast-mae环境的时候。
./forecast-mae/src/model/layers/agent_embedding.py文件中
导入包的时候from natten import NeighborhoodAttention1D
报错,
解决办法,默认安装的是__version__ = "0.14.2"版本的
出现这个问题可能是包损坏所致,
解决办法:可以重新安装 ,可以由低到高版本尝试,重装natten。
pip install natten==0.14.2pip install natten==0.14.4pip install natten==0.14.6
关于natten的介绍:
NATTEN is an extension to PyTorch, which provides the first fast sliding window attention with efficient CUDA kernels. It provides Neighborhood Attention (local attention) and Dilated Neighborhood Attention (sparse global attention, a.k.a. dilated local attention) as PyTorch modules for both 1D and 2D data.
NATTEN是PyTorch的扩展,它提供了第一个快速滑动窗口关注与高效的CUDA内核。它作为PyTorch模块,为1D和2D数据提供邻域注意(局部注意)和扩展邻域注意(稀疏全局注意,又称扩展局部注意)。