Google Cloud ML Engine не может найти локальные TFRecords
Я пытаюсь использовать Google Cloud ML Engine для оптимизации гиперпараметров для моей модели вариационного автоэнкодера, но задание не выполняется, поскольку файлы.tfrecord, которые я указываю для ввода, не найдены. В коде моей модели я передаю train.tfrecords своему входному тензору, как в каноническом примере cifar10, и указываю местоположение train.tfrecords с полным путем.
Релевантная информация:
- JOB_DIR указывает на каталог тренера
- Это изображение показывает мою структуру каталогов
Мой файл setup.py ниже:
from setuptools import find_packages from setuptools import setup REQUIRED_PACKAGES = ['tensorflow==1.3.0', 'opencv-python'] setup( name='trainer', version='0.1', install_requires=REQUIRED_PACKAGES, packages=find_packages(), include_package_data=True, description='My trainer application package.' )
1 ответ
Когда задание выполнено, оно не сможет читать данные с вашего локального компьютера. Самый простой способ сделать файлы TFRecord доступными для вашей работы - скопировать их в GCS, а затем передать местоположение файлов GCS в вашу программу в качестве флагов и использовать эти флаги для настройки ваших читателей и писателей.