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

Question

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

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

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