Существуют ли TPU-совместимые предварительно обученные контрольные точки для классификации изображений? - PullRequest
0 голосов
/ 08 сентября 2018

Я нашел TPU-совместимые предварительно обученные контрольные точки для обнаружения объектов, но не для классификации изображений. Меня особенно интересует Mobilenet_v2. Я бы переключился на другую сеть, если бы такие контрольные точки были доступны, а контрольные точки для Mobilenet_v2 - нет.

1 Ответ

0 голосов
/ 08 сентября 2018

Да, в этих файлах GCS есть контрольная точка Resnet, совместимая с TPU:

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

Вы можете использовать эту контрольную точку, если она соответствует вашим требованиям. Если нет, вы можете выполнить этот урок , чтобы обучить его самостоятельно.

...