Google Cloud ML Engine: не удалось создать версию модели

Question

Я успешно обучил TensorForestEstimator на движке Google Cloud ML, но когда я пытаюсь создать версию модели, я получаю следующую ошибку:

Ошибка создания версии. Обнаружена плохая модель с ошибкой: "Ошибка загрузки модели: не удалось загрузить модель".

Я разворачиваю с tensorflow 1.3, Experiment настроен следующим образом:

def get_experiment_fn(args):
    def _experiment(run_config, hparams):
        return Experiment(
            estimator=TensorForestEstimator(
                params=ForestHParams(
                    num_trees=args.num_trees,
                    max_nodes=10000,
                    min_split_samples=2,
                    num_features=8,
                    num_classes=args.num_projections,
                    regression=True
                ),
                model_dir=args.job_dir,
                graph_builder_class=RandomForestGraphs,
                config=run_config,
                keys_name=None,
                report_feature_importances=True
            ),
            train_input_fn=get_input_fn(
                project_name=args.project,
                data_location=args.train_data,
                dataset_size=args.train_size,
                batch_size=args.train_batch_size
            ),
            train_steps=args.train_steps,
            eval_input_fn=get_input_fn(
                project_name=args.project,
                data_location=args.eval_data,
                dataset_size=args.eval_size,
                batch_size=args.eval_batch_size
            ),
            eval_steps=args.eval_steps,
            eval_metrics=get_eval_metrics(),
            export_strategies=[
                make_export_strategy(
                    serving_input_fn,
                    default_output_alternative_key=None,
                    exports_to_keep=1
                )
            ]
        )
    return _experiment

В чем проблема?

Answer 1

Похоже, Google Cloud ML Engine поддерживает только обслуживающие модели, созданные с использованием tensorflow 1.2.0 и ниже по состоянию на сейчас. Смотрите здесь: https://cloud.google.com/ml-engine/docs/concepts/runtime-version-list

использование --runtime-version 1.2 если возможно. Если вы используете функцию, специфичную для tensorflow 1.3, вам нужно будет разместить свою модель, используя Flask в Google App Engine, пока поддержка ML Engine для tensorflow 1.3 прибывает.