我发现TPU兼容的预训练检查点可用于物体检测,但不能用于图像分类。我对Mobilenet_v2特别感兴趣。如果有这样的检查点可用,而没有用于Mobilenet_v2的检查点,我将切换到另一个网络。
答案 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自己进行培训。