TNT v0.2.0

[Rishit-dagli]

This is an interesting release for the project, including a pre-trained model on ImageNet, reproducibility of paper results, tests, and end-to-end training.

✅ Bug Fixes / Improvements

• Create an end-to-end training example demonstrating how to train a TNT model for image classification through a custom training loop on the TF Flowers dataset (Add a training example #14)
• Pre-trained model to reproduce the paper results have been made available (in this release as well as on TensorFlow Hub)
• Create an off-the-shelf inference example, that highlights how you can directly use the pre-trained model made available
• Unit Tests for the Attention class (Tests for Attention class #19)
• Unit Tests for the main TNT class (Tests for TNT class #20)

Full Changelog: v0.1.0...v0.2.0

---

This discussion was created from the release TNT v0.2.0.