Блог

Главная - Вопросы по программированию - Слой TF keras больше не сохраняется?

Слой TF keras больше не сохраняется?

#tensorflow #tensorflow2.0 #keras-layer #tf.keras #tensorflow-agents

#tensorflow #tensorflow2.0 #keras-layer #tf.keras #tensorflow-агенты

Вопрос:

После недавнего обновления до Tensorflow 2.3 я не могу сохранить слои TF-агентов, я получаю это:

 AttributeError: 'ActorDistributionNetwork' object has no attribute 'save_weights'

Поскольку ActorDistributionNetwork является подклассом tf.keras.layers.Layer , была ли удалена способность отдельных слоев keras сохранять себя? Я ничего не смог найти об этом в изменениях выпуска ни для tensorflow , ни для tf-agents .
Использование model.save_weights не очень удобно для tf-agents , поскольку мне приходится использовать разные комбинации слоев для пользовательского агента.

Комментарии:

1. Почему бы не наследовать от Model вместо Layer ? Это халтурно, но работает.

2. Это был бы вопрос к tf-agents разработчикам, но они решили этого не делать

3. Вы всегда можете обернуть Model вокруг слоя в своем коде.