huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.
4.07k stars 708 forks source link

'SNNMLP' object has no attribute 'module' #256

Closed luoxuanxx closed 7 months ago

luoxuanxx commented 7 months ago

When I train the snn_mlp model, it shows the error: torch.nn.modules.module.ModuleAttributeError: 'SNNMLP' object has no attribute 'module' I did not find the module attribute in the snnnmlp class.