search

Event-AHU / SSTFormer

[PokerEvent Benchmark Dataset & SNN-ANN Baseline] Official PyTorch implementation of "SSTFormer: Bridging Spiking Neural Network and Memory Support Transformer for Frame-Event based Recognition"
16 stars 1 forks source link

Problems running train_SSTFormer.sh #9

Open AmamiyaKarin opened 8 months ago

AmamiyaKarin commented 8 months ago

Traceback (most recent call last): File "tools/train.py", line 136, in main() File "tools/train.py", line 129, in main runner = Runner.from_cfg(cfg) File "……/anaconda3/envs/ECE/lib/python3.8/site-packages/mmengine/runner/runner.py", line 462, in from_cfg runner = cls( File "……/anaconda3/envs/ECE/lib/python3.8/site-packages/mmengine/runner/runner.py", line 429, in init self.model = self.build_model(model) File "……/anaconda3/envs/ECE/lib/python3.8/site-packages/mmengine/runner/runner.py", line 836, in build_model model = MODELS.build(model) File "……/anaconda3/envs/ECE/lib/python3.8/site-packages/mmengine/registry/registry.py", line 570, in build return self.build_func(cfg, args, kwargs, registry=self) File "……/anaconda3/envs/ECE/lib/python3.8/site-packages/mmengine/registry/build_functions.py", line 232, in build_model_from_cfg return build_from_cfg(cfg, registry, default_args) File "……/anaconda3/envs/ECE/lib/python3.8/site-packages/mmengine/registry/build_functions.py", line 121, in build_from_cfg obj = obj_cls(args) # type: ignore File "……/mmaction2/mmaction/models/recognizers/base.py", line 116, in init self.backbone = MODELS.build(backbone) File "……/anaconda3/envs/ECE/lib/python3.8/site-packages/mmengine/registry/registry.py", line 570, in build return self.build_func(cfg, args, **kwargs, registry=self) File "……/anaconda3/envs/ECE/lib/python3.8/site-packages/mmengine/registry/build_functions.py", line 232, in build_model_from_cfg return build_from_cfg(cfg, registry, default_args) File "……/anaconda3/envs/ECE/lib/python3.8/site-packages/mmengine/registry/build_functions.py", line 100, in build_from_cfg raise KeyError( KeyError: 'SSTFormer is not in the mmaction::model registry. Please check whether the value of SSTFormer is correct or it was registered as expected. More details can be found at https://mmengine.readthedocs.io/en/latest/advanced_tutorials/config.html#import-the-custom-module'

WuZongzhen commented 7 months ago

Thank you for your attention! SSTFormer has been added to the SSTFormer/mmaction/models/packages/ init. py file (from .SSTFormer import SSTFormer) , and I did not encounter the issue you mentioned.