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

Question

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

Answer 1

是的，在这些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

如果此检查点满足您的要求，则可以使用它。如果没有，您可以按照本教程自行训练。