Как объединить две модели глубокого обучения ONNX

У меня две модели в формате ONNX. Обе модели похожи (обе являются предварительно обученными моделями глубокого обучения, например, ResNet50). Единственная разница между ними в том, что последние слои оптимизированы / переобучены для разных наборов данных.

Я хочу объединить первые k слоев этих двух моделей, как показано ниже. Это должно повысить эффективность вывода.

Чтобы прояснить мой случай, это примеры из других инструментов машинного обучения для реализации этой функции. Пример. Pytorch, Keras.

Объединить


person Walid Hanafy    schedule 07.08.2020    source источник


Ответы (2)


пакет sclblonnx предоставляет ряд функций более высокого уровня для редактирования графиков ONNX, включая возможность объединения двух подграфов. .

person Maurits Kaptein    schedule 29.04.2021

Вы можете использовать пакет ONNX и предоставляемые им API (https://github.com/onnx/onnx/blob/master/docs/PythonAPIOverview.md) для изменения моделей / графиков.

person Hariharan Seshadri    schedule 11.08.2020