Как установить необучаемые веса? ⇐ Python

[Anonymous]

Метод keras.Model.set_weights, похоже, принимает только обучаемые веса. Необучаемые веса, например веса из слоев нормализации, нельзя импортировать таким образом. Это проблематично, поскольку в Keras с бэкэндом JAX мы можем выполнять только операции без сохранения состояния. То есть; обновить веса отдельно от модели. Поэтому перед сохранением модели с помощью keras.Model.save необходимо загрузить веса. Однако, поскольку необучаемые веса не могут быть загружены (и, следовательно, не сохранены в формате .keras), сохраненная модель будет работать недостаточно эффективно.
Можно ли загрузить/ установить необучаемые веса в модели Keras? В более общем плане, есть ли способ сохранить полную модель, включая необучаемые веса, при использовании JAX в качестве серверной части?

Подробнее здесь: https://stackoverflow.com/questions/792 ... le-weights