Изменить контрольную точку TensorFlow

Я знаю, что есть способ проверить контрольные точки TF (как описано в inspect_checkpoint.py).

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

1 ответ

Решение

Посмотрите на сохранение и загрузку MetaGraph. Там есть несколько примеров; Вы должны быть в состоянии определить модель, сохранить MetaGraph, затем загрузить и изменить ее без переопределения переменных вручную.

Другие вопросы по тегам