VGG на gcp ml-engine работает очень медленно - PullRequest
Новая
       13

VGG на gcp ml-engine работает очень медленно

0 голосов
/ 04 ноября 2018

У меня есть модель на мл-движке google cloud, в основном состоящая из реализации keras нейронной сети VGG19 (даже без полностью связанных слоев в верхней части). Точнее, размер изображения сначала изменяется с помощью tf.image.resize_images, а затем проходит через VGG со средними активациями объектов, рассчитанными в 5 различных слоях.

При запуске из терминала как 

gcloud ml-engine predict \
    --model Features \
    --version v1 \
    --json-instances \
    test_image.json

все время выполнения составляет примерно 17-18 секунд (не при первом запуске).

Похоже, что это намного медленнее, чем следует пройти через VGG. Есть ли у вас какие-либо идеи относительно того, что может быть причиной этого?

1 Ответ

0 голосов
/ 05 ноября 2018

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

У нас есть другие предложения на альфа и бета-этапах (в том числе на GPU в ближайшее время); пожалуйста, свяжитесь с cloudml-feedback@google.com, чтобы запросить доступ, если вы хотите получить доступ.

...