Как предотвратить керас от переименования слоев - PullRequest
2 голосов
/ 16 октября 2019

Когда я заново создаю модель, keras всегда создает новое имя для слоя (conv2d_2 и т. Д.), Даже если я переопределяю модель. Как сделать, чтобы керасы использовали одно и то же имя каждый раз, когда я запускаю его, без перезапуска ядра.

1 Ответ

1 голос
/ 17 октября 2019

При использовании Tensorflow (1.X) в качестве бэкэнда, когда вы добавляете новый слой в какую-либо модель, имя слоя, если оно не задано вручную, будет установлено на имя по умолчанию для этого слоя плюс инкрементный индекс. в конце.

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

В TF 1.X это делается с помощью tf.reset_default_graph(). В TF 2.0 вы можете сделать это через API совместимости v1: tf.compat.v1.reset_default_graph() (последний также решит некоторые предупреждения об устаревании, которые вы можете получить с последними версиями TF 1.X)

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...