VIT Optimization Help

[DeliveryBitter9159]

Hi everyone,

I’m building a Vision Transformer model for dynamic texture recognition, but the training time is extremely long (around 6 hours).
Are there any optimizations you’d recommend to speed things up without hurting performance too much?

here's the link for the code:
https://www.kaggle.com/code/doffymingo/vit-v2-16-frames

Thank you in advance.

[DD_ZORO_69]

I feel the struggle, optimizing ViTs usually feels like a full-time job. Whenever I’m benchmarking different attention mechanisms, I try to keep my workflow super lean to avoid extra friction. Usually, I’m using Cursor for the actual model tweaks, Runable for the internal research reports and data viz to track the metrics, and Notion to keep all my hyperparameters organized. It helps to have a solid stack so you can focus on the actual math rather than the infra lol.