Поскольку ONNX поддерживает ограниченные модели, я попытался выполнить это преобразование, назначив параметры напрямую, но полученная модель тензорного потока не смогла показать желаемую точность. Подробности описаны следующим образом:
- Исходная модель обучена Ленету по набору данных MNIST.
- Сначала я извлек каждый модуль и его параметры с помощью model.named_parameters () и сохранил их всловарь, в котором ключом является имя модуля, а значением являются параметры
- Затем я построил и инициировал модель тензорного потока с той же архитектурой
- Наконец, я назначаю параметры каждого слоя модели Pytroch длямодель тензорного потока
Однако точность полученной модели тензорного потока составляет всего около 20%. Таким образом, мой вопрос заключается в том, возможно ли преобразовать модель pytorch этим методом? Если да, какова возможная проблема, приводящая к плохому результату? Если нет, пожалуйста, объясните причины.
PS: предположим, что процедура назначения правильная.