«обрезать» граф onnx от его констант (инициализаторов) - PullRequest
Новая
       9

«обрезать» граф onnx от его констант (инициализаторов)

0 голосов
/ 06 ноября 2018

У меня есть граф / модель onnx, в которой есть большие константы, поэтому загрузка и анализ ее занимает много времени. Могу ли я «удалить» данные с графика, чтобы проверить узлы графика без его данных?

1 Ответ

0 голосов
/ 04 декабря 2018

Инициализатор является одним из полей в GraphProto. Вы должны быть в состоянии очистить поле инициализатора простым скриптом Python. Я не тестировал следующий код, но должен выглядеть примерно так: 

import onnx

def clear_initializer(model_path):
    model = onnx.load_model(model_path)
    model.graph.ClearField('initializer')
    onnx.save_model(model)

ссылка: https://developers.google.com/protocol-buffers/docs/reference/python/google.protobuf.message.Message-class https://github.com/onnx/onnx/blob/2e7099ee7c37b196c197c9a084a97698a41da232/onnx/init.py

...