TensorFlow Hub new Models thread

This thread is aimed to highlight new released models on TensorFlow Hub

11 Likes

New BERT model available:

MobileBERT is a compact task-agnostic BERT for Resource limited devices.
This version is an upgrade on the previous version, compatible with Preprocess model and also provides trainable Masked Language Model!

English uncased version
Multi language cased version

4 Likes

Mobilenet V3 is now available!

There are 8 new models, 4 [large, small]x[1.0, 0.75] for classification and their respective 4 feature vectors

They have also their TFLite and TF.js versions available

You can find links to all of them on this collection

5 Likes

This is indeed great. Would appreciae a TFLite variant as well. I did try to make it work (an earlier version) but the resulting model was spurious.

Here’s where I have logged all the progress: GitHub - sayakpaul/BERT-for-Mobile: Compares the DistilBERT and MobileBERT architectures for mobile deployments..

My idea was to author a blog post on “Optimizing different BERTs for Mobile” or something like that but didn’t happen because of lack of proper comparison metrics :frowning:

4 Likes

Next-Generation Pose Detection with MoveNet

The model has two versions: Lighting and Thunder and both have the regular SavedModel version and also the TensorFlow.js version ready to use.

They are quite fast and with great accuracy. You can try them out on this guide:
[ MoveNet: Ultra fast and accurate pose detection model]
(MoveNet: Ultra fast and accurate pose detection model.)

4 Likes

Recently the FRILL models were published on TFHub!

These are small, fast and great for on-device human non-semantic speech sound classification like detecting when someone is speaking with or without a mask and related tasks!

https://tfhub.dev/s?q=nonsemantic-speech-benchmark%2Ffrill

if you want even more information there’s also this nice blog post explaining more details on how the model is trained: Google AI Blog: FRILL: On-Device Speech Representations using TensorFlow-Lite

1 Like

The new EfficientNet V2 models are all available on TensorFlow Hub!

go check it out and let us know what you build with them!

1 Like