Форма [-1, 1] имеет отрицательные размеры

Я работаю над одной проблемой классификации классов по тензорному потоку. Обучение закончено, и я сохранил свою модель. Когда я пытаюсь вызвать прогноз с помощью облака Google, я получаю эту ошибку:

{
  "error": "Prediction failed: Error during model execution: AbortionError(code=StatusCode.INVALID_ARGUMENT, details=\"Shape [-1,1] has negative dimensions\n\t [[
Node: outputs = Placeholder[_output_shapes=[[-1,1]], dtype=DT_FLOAT, shape=[?,1], _device=\"/job:localhost/replica:0/task:0/cpu:0\"]()]]\")"
}

Я определил свою подпись предсказания для выходных данных как build_tensor_info выходной переменной, которую я определил следующим образом:

Y = tf.placeholder("float", [None, n_classes], name = 'outputs')

Обучение и тестирование отлично работают на моем ноутбуке и в облаке Google. Только когда я призываю к предсказаниям, я получаю эту ошибку. Я использую этот код для вызова прогнозов:

gcloud ml-engine predict --model ${MODEL_NAME} --json-instances request.json

Я считаю, что это простая проблема, которую мне не хватает в отношении того, как работает форма в python / tenorflow.

Кто-нибудь еще имел эту проблему раньше?

0 ответов

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