apapiu / mamba_small_bench Goto Github PK

Dear author I have been following the source code you provided. I am fairly new to Deep Learning development.

About the Trainer, it shows that it has unexpected argument 'lr'. So it this code executable and bug free? Or do you have another hidden script

Besides, for the code cifar_10.py on the repo, does it only demonstrate the training and evaluation on Mamba transformer?
How did your graph appear 2 results?

I have tried to contact you via this email [email protected] but in vain, hope that I can contact you for more info.

This is my email: [email protected]

Hello authors!

I've observed that you've attempted to train Mamba on image classification tasks. I'm curious: have you encountered the segmentation fault error during this process?

I've been training Mamba on ImageNet1K, but it appears to be quite unstable. After several epochs, I encounter a segmentation fault that interrupts the training process. Have you experienced this issue as well?