Skip to content

hello, can you help me solve this problem?name 'Mamba_FFT' is not defined #12

@liubaoning111

Description

@liubaoning111

Traceback (most recent call last):
File "/home/liubn/0-3/T-Mamba-main/./train.py", line 409, in
main()
File "/home/liubn/0-3/T-Mamba-main/./train.py", line 381, in main
model, optimizer, lr_scheduler = models.get_model_optimizer_lr_scheduler(params)
File "/home/liubn/0-3/T-Mamba-main/lib/models/init.py", line 150, in get_model_optimizer_lr_scheduler
model = TMamba2D(in_channels=opt["in_channels"], classes=opt["classes"], scaling_version=opt["scaling_version"], input_size=opt['resize_shape'], high_freq=opt['high_frequency'], low_freq=opt['low_frequency'])
File "/home/liubn/0-3/T-Mamba-main/lib/models/TMamba2D.py", line 217, in init
DownsampleWithDfs2D(
File "/home/liubn/0-3/T-Mamba-main/lib/models/TMamba2D.py", line 398, in init
self.dfs = DenseFeatureStack(
File "/home/liubn/0-3/T-Mamba-main/lib/models/TMamba2D.py", line 357, in init
self.Tim_Block.append(Tim_Block(growth_rate, fused_add_norm=True, residual_in_fp32=True,
File "/home/liubn/0-3/T-Mamba-main/lib/models/TMamba2D.py", line 61, in init
mixer_cls = partial(Mamba_FFT, layer_idx=None, bimamba_type=bimamba_type, high_freq=high_freq, low_freq=low_freq, **ssm_cfg, **factory_kwargs)
NameError: name 'Mamba_FFT' is not defined

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions