myxmhy/ConvMamba

4

stars

7

commits

Python

primary language

Jun 10, 2024

updated

README

ConvMamba

Overview

Code Structures
  • core/ core training plugins and metrics.
  • methods/ contains training methods for various methods.
  • Model/ contains the control files for various methods and is used to store the model and training results.
  • models/ contains the main network architectures of various methods.
  • modules/ contains network modules and layers.
  • runfiles/ contains various startup scripts.
  • utils/ contains a variety of utilities for model training, model modules, parsers, etc.
  • DataDefine.py is used to get the dataset and make a dataloader.
  • modelbuild.py is used to build and initialize the model.
  • modeltrain.py is used to train, validate, test and inference about model.
  • main.py is the main function that runs the program.
  • inference.py is used for model inference.
  • test.py is used for model test.

Overall methodology and relevant experiments and discussions

Matrices B and C in traditional SSM and selective SSM

Structure of Mamba block

Contributors

mhy9989

7 commits

myxmhy/ConvMamba

4

stars

7

commits

Python

primary language

Jun 10, 2024

updated

README

ConvMamba

Overview

Code Structures
  • core/ core training plugins and metrics.
  • methods/ contains training methods for various methods.
  • Model/ contains the control files for various methods and is used to store the model and training results.
  • models/ contains the main network architectures of various methods.
  • modules/ contains network modules and layers.
  • runfiles/ contains various startup scripts.
  • utils/ contains a variety of utilities for model training, model modules, parsers, etc.
  • DataDefine.py is used to get the dataset and make a dataloader.
  • modelbuild.py is used to build and initialize the model.
  • modeltrain.py is used to train, validate, test and inference about model.
  • main.py is the main function that runs the program.
  • inference.py is used for model inference.
  • test.py is used for model test.

Overall methodology and relevant experiments and discussions

Matrices B and C in traditional SSM and selective SSM

Structure of Mamba block

Contributors

mhy9989

7 commits

Languages

Python

94.1%

Shell

5.9%