Использование умных людей только с весами модели и без класса модели

Question

Я использую предварительно подготовленную модель, созданную кем-то другим, они только выпустили веса моделей. В настоящее время я импортирую весовые коэффициенты модели в свой график и называю их тензорными именами. Тем не менее, это кажется несовместимым с кодом умных людей, который, кажется, требует модельный объект, который имеет метод предиката.

Есть ли обходной путь для этого, который не требует от меня переписывания большинства атак cleverhans, потому что у меня нет класса модели и метода прогнозирования?

Answer 1

То, что вы описываете, должно быть возможным, но может быть несколько интенсивным по ресурсам, потому что это может воссоздать график несколько раз. По сути, вы можете реализовать класс модели CleverHans, который принимает контрольную точку графа в методе init. get_logits или же fprop Метод должен взять входной тензор и загрузить график, чтобы получить соответствующий выходной тензор, выполнив некоторые графические операции, чтобы заменить входной тензор графа контрольных точек своим собственным тензором: см. input_map аргумент в `tf.import_graph_de: https://www.tensorflow.org/api_docs/python/tf/graph_util/import_graph_def