Как квантовать все узлы, кроме одного?

Я использую инструмент Graph Transform Tool для вычисления квантования графика, используя

input_names = ["prefix/input"]
output_names = ["final_result"]


transforms1 = ["strip_unused_nodes","fold_constants(ignore_errors=true)",  "fold_batch_norms",  "fold_old_batch_norms","quantize_weights" ]

transformed_graph_def = TransformGraph(graph.as_graph_def(), input_names,output_names, transforms1)

Я использую параметр quantize_weights для квантования весов в графе, я знаю, что некоторые узлы могут оставаться неквантованными, изменяя пороговое значение minimum_size в quantize_weights, поэтому оставлять некоторые узлы неквантованными, безусловно, возможно.

Я хочу квантовать веса всех узлов, кроме определенного узла с именем K или набором узлов, которые имеют имя в K (set). Как это можно достичь?

Ответы