Компиляция Sagemaker Neo с Dynami c Размер входных данных - PullRequest
Новая
       48

Компиляция Sagemaker Neo с Dynami c Размер входных данных

0 голосов
/ 21 марта 2020

Я пытаюсь скомпилировать модель PyTorch в малину p ie 3. Модель, которую я использую, не имеет никаких ограничений в отношении ширины и высоты. Sagemaker Neo принимает параметр Data input configuration, который выглядит так, как будто его необходимо исправить из документации и примеров. 

Data Input Configuration:  
Amazon SageMaker needs to know the what the shape of the data matrix is
{"data": [1, 3, 224, 224]}

Приведенный выше пример можно найти здесь .

Мне нужно что-то вроде следующего: 

Data Input Configuration:  
Amazon SageMaker needs to know the what the shape of the data matrix is
{"data": [1, 3, *, *]}

, введя выше, выдает следующее сообщение об ошибке: 

JSON.parse: unexpected character at line 1 column 17 of the JSON data

и фиксируя высоту и ширину, получаем следующее сообщение: 

ClientError: InputConfiguration: Invalid PyTorch model or input-shape mismatch. Make sure that inputs are lexically ordered and of the correct dimensionality.

Можно ли вообще указать, что я хочу 1 изображение, с 3 каналами и переменной высотой и шириной?

...