search

mlcommons / mobile_models

MLPerf™ Mobile models
https://mlcommons.org/en/groups/inference-mobile/
Apache License 2.0
24 stars 10 forks source link

MobileBert with TFlite GPU delegate #19

Closed Mo-Abdelgawad closed 3 years ago

Mo-Abdelgawad commented 3 years ago

Could you please add information (tflite conversion or the training process) on how to reproduce mobilebert tflite model for mobile gpu ( https://github.com/mlcommons/mobile_models/blob/main/v0_7/tflite/mobilebert_float_384_gpu.tflite)?

I need to experiment with different mobilebert configurations on mobile gpus with TFlite GPU delegate for my research, and this is the only model I have found that can actually run with TFlite GPU delegate!

Thank you.

mcharleb commented 3 years ago

@Mo-Abdelgawad Have a look at https://github.com/mlcommons/mobile_open/tree/main/language/bert