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 в вашу программу в качестве флагов и использовать эти флаги для настройки ваших читателей и писателей.

Другие вопросы по тегам