gamba - Pytorch
Project description
Gamba
Implementation of "GAMBA: MARRY GAUSSIAN SPLATTING WITH MAMBA FOR SINGLE-VIEW 3D RECONSTRUCTION" in PyToch
install
$ pip intall gamba
usage
import torch
from gamba.main import Gamba
# Forward pass of the GambaDecoder module.
x = torch.randn(1, 1000, 512)
# Model
model = Gamba(
dim=512,
d_state=512,
d_conv=512,
n=16384,
depth=3
)
# Out
out = model(x)
print(out)
License
MIT
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
gamba_torch-0.0.1.tar.gz
(4.5 kB
view hashes)
Built Distribution
Close
Hashes for gamba_torch-0.0.1-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | d3e226af2acc2d4b49776302f114fa68244211d282af225727cbc2003961998c |
|
MD5 | e64748828a0adc01fd45e5f7cb9b6d4c |
|
BLAKE2b-256 | 6a9b8f79f9d4bf5594d0d44161d69336c918b423b265a42e16715d1cbde88f9c |