Ошибка при преобразовании модели pytorch в core-ml - PullRequest
6 голосов
/ 20 мая 2019
C = torch.cat((A,B),1)

форма тензоров:

A is (1, 128, 128, 256)
B is (1, 1, 128, 256)

Ожидаемое C значение (1, 129, 128, 256)

Этот код работает на pytorch, но при преобразовании в core-ml он даетмне ниже ошибка:

"Error while converting op of type: {}. Error message: {}\n".format(node.op_type, err_message, )
TypeError: Error while converting op of type: Concat. Error message: unable to translate constant array shape to CoreML shape"
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...