Thank you very much for your code, which has helped me a lot! But I have a question for you. Thank you

[liubaoning111]

/home/liubn/anaconda3/envs/swin_umamba/lib/python3.10/site-packages/torchvision/io/image.py:13: UserWarning: Failed to load image Python extension: '/home/liubn/anaconda3/envs/swin_umamba/lib/python3.10/site-packages/torchvision/image.so: undefined symbol: _ZN3c1017RegisterOperatorsD1Ev'If you don't plan on using image functionality from torchvision.io, you can ignore this warning. Otherwise, there might be something wrong with your environment. Did you have libjpeg or libpng installed before building torchvision from source? warn( tvdcn is installed, using it for deformable convolution Traceback (most recent call last): File "/home/liubn/anaconda3/envs/swin_umamba/lib/python3.10/site-packages/natten/functional.py", line 32, in from natten import libnatten # type: ignore ImportError: /home/liubn/anaconda3/envs/swin_umamba/lib/python3.10/site-packages/natten/libnatten.cpython-310-x86_64-linux-gnu.so: undefined symbol: _ZN2at4cuda19getDevicePropertiesEl

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "/home/liubn/0-149liubaoning/43-AgileFormer-main/train.py", line 12, in from networks.agileFormer_2d import AgileFormer2D as ViT_seg File "/home/liubn/0-149liubaoning/43-AgileFormer-main/networks/agileFormer_2d.py", line 3, in from .agileFormer_sys_2d import AgileFormerSys2D File "/home/liubn/0-149liubaoning/43-AgileFormer-main/networks/agileFormer_sys_2d.py", line 9, in from .nat_2d import NeighborhoodAttention2D File "/home/liubn/0-149liubaoning/43-AgileFormer-main/networks/nat_2d.py", line 30, in from natten.functional import na2d_av, na2d_qk_with_bias File "/home/liubn/anaconda3/envs/swin_umamba/lib/python3.10/site-packages/natten/init.py", line 24, in from .functional import ( File "/home/liubn/anaconda3/envs/swin_umamba/lib/python3.10/site-packages/natten/functional.py", line 34, in raise ImportError( ImportError: Failed to import NATTEN's CPP backend. This could be due to an invalid/incomplete install. Please uninstall NATTEN (pip uninstall natten) and re-install with the correct torch build: shi-labs.com/natten .