Игнорировать определенные переменные при сохранении оценщика - PullRequest
0 голосов
/ 07 июня 2019

У меня есть собственный оценщик, который использует vgg-модель для сопоставления объектов (мне нужны градиенты при активации vgg) - поэтому vgg должен быть частью моей модели (model_fn).Эти переменные vgg не оптимизированы и восстанавливаются с помощью пользовательской функции (переменные vgg назначаются из отдельного каталога, поэтому существует одна для всех моделей).Я не хочу загружать или сохранять эти переменные в моих контрольных точках.Как я могу сделать это с помощью tf.estimator.Estimator?Я знаю, что могу фильтровать свои переменные с помощью WarmStartSettings, но это cumberstone, и переменные по-прежнему сохраняются.

Существует похожая проблема: Как удалить непереносимые переменные при сохранении контрольной точки с помощью оценки тензорного потока? Но я не могу найти способ изменить заставку (похоже, этот парень спас модель вручную?).

...