Загрузка предварительно обученной модели Resnet в формат модели Cleverhans
Я пытаюсь загрузить предварительно обученную модель ResNet из MadryLab CIFAR-10 вызов в CleverHans рассчитать передачу атаки.
Однако восстановление сохраненных моделей в model_zoo.madry_lab_challenges.cifar10_model.ResNet объект не работает. Похоже, что сначала восстанавливается нормально, но когда я пытаюсь использовать модель, я получаю сообщение об ошибке:
Попытка использовать неинициализированное значение ResNet/unit_3_1/residual_only_activation/BatchNorm/moving_mean
Самый простой способ воспроизвести эту ошибку - просто запустить attack_model.py пример включен в CleverHans Вот:
Он сталкивается с той же ошибкой после загрузки модели, когда он пытается использовать ее, на обоих adv_trained а также naturally_trained моделей.
Есть ли решение этой проблемы?
Похоже, другой вариант заключается в использовании cleverhans.model.CallableModelWrapper вместо этого, но я не смог найти пример того, как это использовать.