More Thoughts on Mamba: Predicting the view count of a Youtube video from its thumbnail with VisionMamba

Follow the full discussion on Reddit.
For fun, I wrote an article about my project to predict the view count of a video from its thumbnail, using a variation of Vision Transformer that uses Mamba as a backbone instead. It's a really nice model to use. Its memory-lightness means I can use it as a GPU-poor person, which is awesome. This is the first post in a series, I plan to scale up the dataset in the future, incorporate text embeddings, and then finally use it with RLHF to finetune a stable diffusion model.

Comments

There's unfortunately not much to read here yet...

Discover the Best of Machine Learning.

Ever having issues keeping up with everything that's going on in Machine Learning? That's where we help. We're sending out a weekly digest, highlighting the Best of Machine Learning.

Join over 900 Machine Learning Engineers receiving our weekly digest.

Best of Machine LearningBest of Machine Learning

Discover the best guides, books, papers and news in Machine Learning, once per week.

Twitter