Sigma代码复现

ganchuanl

已于 2024-07-27 15:08:27 修改

阅读量408

点赞数 3

文章标签： python 人工智能深度学习

于 2024-07-27 14:58:19 首次发布

本文链接：https://blog.csdn.net/ganchuanl/article/details/140735572

版权

Traceback (most recent call last):
File "/home/gcl/project/CSCA-main/train.py", line 79, in <module>
    trainer.setup()
File "/home/gcl/project/CSCA-main/utils/regression_trainer.py", line 98, in setup
    self.model = segmodel(args, criterion=criterion, norm_layer=BatchNorm2d)
File "/home/gcl/project/CSCA-main/models/builder.py", line 62, in __init__
    from .encoders.dual_vmamba import vssm_small as backbone
File "/home/gcl/project/CSCA-main/models/encoders/dual_vmamba.py", line 12, in <module>
    from models.encoders.vmamba import Backbone_VSSM, CrossMambaFusionBlock, ConcatMambaFusionBlock
File "/home/gcl/project/CSCA-main/models/encoders/vmamba.py", line 32, in <module>
    import selective_scan_cuda_core as selective_scan_cuda
ModuleNotFoundError: No module named 'selective_scan_cuda_core'

Traceback (most recent call last):
File "/home/gcl/project/CSCA-main/train.py", line 80, in <module>
    trainer.train()
File "/home/gcl/project/CSCA-main/utils/regression_trainer.py", line 139, in train
    self.train_eopch()
File "/home/gcl/project/CSCA-main/utils/regression_trainer.py", line 166, in train_eopch
    outputs = self.model(inputs, self.dataset)
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
    return forward_call(*args, **kwargs)
File "/home/gcl/project/CSCA-main/models/builder.py", line 155, in forward
    out = self.encode_decode(inputs)
File "/home/gcl/project/CSCA-main/models/builder.py", line 135, in encode_decode
    x = self.backbone(rgb, modal_x)
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
    return forward_call(*args, **kwargs)
File "/home/gcl/project/CSCA-main/models/encoders/dual_vmamba.py", line 110, in forward
    out = self.forward_features(x_rgb, x_e)
File "/home/gcl/project/CSCA-main/models/encoders/dual_vmamba.py", line 100, in forward_features
    cross_rgb, cross_x = self.cross_mamba[i](out_rgb.permute(0, 2, 3, 1).contiguous(), out_x.permute(0, 2, 3, 1).contiguous()) # B x H x W x C
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
    return forward_call(*args, **kwargs)
File "/home/gcl/project/CSCA-main/models/encoders/vmamba.py", line 1870, in forward
    return self._forward(x_rgb, x_e)
File "/home/gcl/project/CSCA-main/models/encoders/vmamba.py", line 1858, in _forward
    x_rgb_cross, x_e_cross = self.op(x_rgb, x_e)
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
    return forward_call(*args, **kwargs)
File "/home/gcl/project/CSCA-main/models/encoders/vmamba.py", line 1633, in forward
    y_rgb, y_e = self.CMA_ssm(x_rgb_conv, x_e_conv)
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1518, in _wrapped_call_impl
    return self._call_impl(*args, **kwargs)
File "/home/gcl/anaconda3/envs/vim/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1527, in _call_impl
    return forward_call(*args, **kwargs)
File "/home/gcl/project/CSCA-main/models/encoders/vmamba.py", line 1509, in forward
    selective_scan = selective_scan_fn_v1
NameError: name 'selective_scan_fn_v1' is not defined. Did you mean: 'selective_scan'?