Получение ошибки при выводе конечной точки sagemaker

Question

Я создал учебную работу в Sagemaker с собственным кодом обучения и логического вывода, используя среду MXNet. Я могу успешно тренировать модель и создать конечную точку. Но, выводя модель, я получаю следующую ошибку:

‘ClientError: Произошла ошибка (413) при вызове операции InvokeEndpoint: длина содержимого HTTP превысила 5246976 байт.’

Из моих исследований я понял, что ошибка связана с размером изображения. Форма изображения (480, 512, 3). Я тренировал модель с изображениями одинаковой формы (480, 512, 3).

Когда я изменил размер изображения до (240, 256), ошибка исчезла. Но в процессе обучения модели с изображениями размера (480, 512) вырабатывается еще одна ошибка - «форма, несовместимая в свертке».

Я не понял, почему я получаю эту ошибку при выводе. Разве мы не можем использовать изображения большего размера, чтобы сделать вывод о модели? Любые предложения будут полезны

Спасибо, Харатхи

Answer 1

Я думаю, что это связано с POST-ответом SageMaker, имеющим ограничение по размеру. Я не видел опцию конфигурации конечной точки для этого в документации.

Вы сможете установить ограничение размера вручную, если вы сделаете докер для вашей модели, а затем измените client_max_body_size, например, здесь, в примере принятия решений: https://github.com/awslabs/amazon-sagemaker-examples/blob/master/advanced_functionality/scikit_bring_your_own/container/decision_trees/nginx.conf#L23