Как я могу сохранить и восстановить график, который использует Tensorflow Hub Module

Question

Я ищу рабочий пример, где я использую модуль изображения TF Hub, настраиваю его для пользовательской задачи, а также обратно распространяю к модулю концентратора. Затем сохраните полученную модель и перезагрузите ее для оценки.

Я посмотрел на https://github.com/tensorflow/hub/blob/master/examples/image_retraining/retrain.py, но я не думаю, что он обучает модуль.

Answer 1

Это просто работает как обычно при восстановлении модели из контрольной точки.

• Если ваша модель восстанавливается из контрольной точки, то переменные модуля также будут восстановлены из нее (а незагружается из исходного модуля).

• Единственный раз, когда переменные модуля устанавливаются на исходные значения, это когда запускаются их инициализаторы.