#952 · Primary category: Education & Research

OmniNet

artificial-intelligence deep-learning image-captioning machine-learning multimodal-learning multitask-learning neural-network nlp transformer video-recognition

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