Различия форматов параметров моделей M XNet и Tensorflow - PullRequest
0 голосов
/ 15 февраля 2020

Я пытаюсь создать набор тестов для Tensorflow Lite, который имитирует реализованный дизайн M XNet.

Полагаю, самый простой или наименее потный способ - заставить модель TFLite загружаться в тот же формат, который использует M XNet, и использовать оставшуюся часть тестового сценария как есть.

Я пытался найти хороший источник информации, но, похоже, нет информации о деталях реализации между различными библиотеками.

Оригинальная реализация M XNet:

sym, arg_params, aux_params = mx.model.load_checkpoint(prefix, epoch)

Реализация Tensorflow Lite:

# Load TFLite model and allocate tensors.
interpreter = tf.lite.Interpreter(model_path="converted_model.tflite")
interpreter.allocate_tensors()

# Get input and output tensors.
input_details = interpreter.get_input_details()
output_details = interpreter.get_output_details()

Мой вопрос заключается в том, как превратить реализацию Tensorflow Lite в формат M XNet реализация, если есть какие-либо документы, это было бы очень полезно.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...