MachRepo

@tridao I tried to use MLP instead and increased the size up to 500 and the model worked, what might be the problem ?

@albertfgu The mamba model is around 500k params and the mlp I used for test had 2 million parameters and still worked perfectly fine. Excuse my ignorance but how to...

@albertfgu Hello again Mr. Gu, Thank you very much for your assistance. I indeed added a print statement inside the module, and got it printed when I used the module....