Have I written custom code (as opposed to using a stock example script provided in MediaPipe)
Yes
OS Platform and Distribution
MacOS & Linux
Python Version
3.10.12
MediaPipe Model Maker version
0.2.1.4
Task name (e.g. Image classification, Gesture recognition etc.)
Model Training
Describe the actual behavior
I am unable to run the MobileBERT model training on GPU & TPU with proper GPU & TPU runtime, it pick tf.config.list_physical_devices() only as CPU
Describe the expected behaviour
Expected to run on all accelerators
Standalone code/steps you may have used to try to get what you need
-- coding: utf-8 -- """training_gpu.ipynb Automatically generated by Colab. Original file is located at https://colab.research.google.com/drive/10GHWGPAZJKM8RbXCpbBt2mit_gs3txbn?resourcekey=0-f-FpXhUgIbZbFHyz0sEC0g """ # @title Pre Import tensorflows import tensorflow as tf assert tf.version.startswith('2') tf.get_logger().setLevel('ERROR') import warnings warnings.filterwarnings('ignore') warnings.filterwarnings('ignore', category=UserWarning) tf.config.list_physical_devices() # @title Add Imports !pip install mediapipe-model-maker # @title Import tensorflows import tensorflow as tf assert tf.version.startswith('2') tf.get_logger().setLevel('ERROR') import warnings warnings.filterwarnings('ignore') warnings.filterwarnings('ignore', category=UserWarning) tf.config.list_physical_devices() from mediapipe_model_maker import text_classifier
Other info / Complete Logs
