Simba

graph A simpler Pytorch + Zeta Implementation of the paper: "SiMBA: Simplified Mamba-based Architecture for Vision and Multivariate Time series"

install

$ pip install simba-torch

usage

import torch 
from simba_torch.main import Simba

# Forward pass with images
img = torch.randn(1, 3, 224, 224)

# Create model
model = Simba(
    dim = 4,                # Dimension of the transformer
    dropout = 0.1,          # Dropout rate for regularization
    d_state=64,             # Dimension of the transformer state
    d_conv=64,              # Dimension of the convolutional layers
    num_classes=64,         # Number of output classes
    depth=8,                # Number of transformer layers
    patch_size=16,          # Size of the image patches
    image_size=224,         # Size of the input image
    channels=3,             # Number of input channels
    # use_pos_emb=True # If you want
)

# Forward pass
out = model(img)
print(out.shape)

License

MIT

Todo

Add paper link
Add citation bibtex
cleanup

Metadata

Release files for simba-torch 0.0.5

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for simba-torch 0.0.5
File	Size	Uploaded
simba_torch-0.0.5.tar.gz	5.7 kB	Mar 26, 2024	Details

Built distribution (wheel)

Table of built distributions (wheels) for simba-torch 0.0.5
File	Interpreter	ABI	Platform	Reset
simba_torch-0.0.5-py3-none-any.whl 6.0 kB Mar 26, 2024	Python 3	none	any	Details

Total release size: 11.7 kB

Release files / simba_torch-0.0.5.tar.gz

Download URL	simba_torch-0.0.5.tar.gz
Size	5.7 kB
Tags	Source
SHA-256 checksum How to use checksums	`e619f4defed636bfe3d7218ede72c1dba003a0558a8372dda3a0e0bbd94240a2`
BLAKE2b-256 checksum How to use checksums	`b1bc7f22a7bdd4166d258bea69751d8d09dc150f1a6eb885fb0a3d96b1c938da`
Upload date	Mar 26, 2024
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`poetry/1.3.2 CPython/3.11.0 Darwin/23.3.0`

Release files / simba_torch-0.0.5-py3-none-any.whl

Download URL	simba_torch-0.0.5-py3-none-any.whl
Size	6.0 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`46ac9b3296780b001b81ad79542bf14290683acbb600a99d130b756d4d470284`
BLAKE2b-256 checksum How to use checksums	`686841bc77be42bf45adc262e91ab96eafeefa90a07a4b3d3d8aa9461a5e9df2`
Upload date	Mar 26, 2024
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`poetry/1.3.2 CPython/3.11.0 Darwin/23.3.0`