#952 · Primary category: Education & Research
OmniNet
Official Pytorch implementation of "OmniNet: A unified architecture for multi-modal multi-task learning" | Authors: Subhojeet Pramanik, Priyanka Agrawal, Aman Hussain
Project last updated:10/31/20
GitHub Stars
514
Forks
59
Contributors
3
License
Apache-2.0
Why we included this project
OmniNet is a compact codebase for anyone curious about training one model on multiple tasks across text, images, and video. It is the code released by the OmniNet paper's authors, so the pretrained weights, dataset download script, and training commands reproduce the published results rather than being a loosely related third-party port. The architecture routes each input domain through its own neural peripheral into a shared Transformer-based processor, and that shared core is what lets a single model handle POS tagging, image captioning, visual question answering, and video activity recognition at once. The repo also demonstrates multi-GPU hogwild training across heterogeneous tasks and includes experimental zero-shot prediction for video captioning and video QA, which makes it useful for teaching or digging into unified multimodal architectures. Just keep in mind it has not been maintained since 2020 and is not built for production deployment.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
prompts.chat
f.k.a. Awesome ChatGPT Prompts. Share, discover, and collect prompts from the community. Free and open source — self-host for your organization with complete privacy.
JavaGuide
Java Interview & Backend General Interview Guide, covering computer fundamentals, databases, distributed systems, high concurrency, system design, and AI application development.
system-prompts-and-models-of-ai-tools
A curated collection of system prompts, internal tools, and AI models from popular AI assistants and coding agents.
30-seconds-of-code
Coding articles to level up your development skills
generative-ai-for-beginners
21 Lessons, Get Started Building with Generative AI