shahidalihakro

I found that I was testing the code on the wrong dataset. After switching to the correct dataset ([ImageNet1K validation set](https://www.kaggle.com/datasets/sautkin/imagenet1kvalid)), everything is working as expected now.

> I had raised this issue a while back, [#571](https://github.com/state-spaces/mamba/issues/571) . Please let me know if you figure out anything. Sure, I will let you know if I can figure...