-
Notifications
You must be signed in to change notification settings - Fork 11
Description
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