我正在按照https://towardsdatascience.com/creating-your-own-object-detector-ad69dda69c85
的指南来训练对象检测模型在Google Colab上,我可以执行以下操作,并且可以利用GPU
python train.py --train_dir=training/ --pipeline_config_path=training/ssd_mobilenet_v1_0.75_depth_quantized_300x300_coco14_sync.config
我现在想使用TPU进行训练,但这显然不只是开箱即用。运行train.py的速度很慢,似乎仅在使用CPU。我该如何实现?