使用TPU在Google Colab上训练对象检测模型

时间:2019-12-06 10:06:21

标签: tensorflow google-colaboratory

我正在按照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。我该如何实现?

0 个答案:

没有答案