是否有TPU兼容的预训练检查点进行图像分类?

时间:2018-09-07 21:09:26

标签: tensorflow google-cloud-tpu

我发现TPU兼容的预训练检查点可用于物体检测,但不能用于图像分类。我对Mobilenet_v2特别感兴趣。如果有这样的检查点可用,而没有用于Mobilenet_v2的检查点,我将切换到另一个网络。

1 个答案:

答案 0 :(得分:0)

是的,这些GCS文件中有一个TPU兼容的Resnet检查点:

gs://cloud-tpu-artifacts/resnet/resnet-nhwc-2018-02-07/checkpoint
gs://cloud-tpu-artifacts/resnet/resnet-nhwc-2018-02-07/model.ckpt-112603.data-00000-of-00001
gs://cloud-tpu-artifacts/resnet/resnet-nhwc-2018-02-07/model.ckpt-112603.index
gs://cloud-tpu-artifacts/resnet/resnet-nhwc-2018-02-07/model.ckpt-112603.meta

如果此检查点满足您的要求,则可以使用它。如果没有,您可以按照this tutorial自己进行培训。