Как заменить существующие узлы графа в Tensorflow?

#python #tensorflow #graph #google-colaboratory

#python #tensorflow #График #google-colaboratory

Вопрос:

У меня уже есть предварительно подготовленный граф.Я выполнил некоторые вычисления для существующих значений веса, чтобы получить объединенные веса batchnorm (у меня есть объединенные веса batchnorm и свертки). Используя свой код, я смог сохранить эти новые веса в файлах. Теперь я хотел заменить эти существующие веса новыми весами на графике. Как это можно сделать? Любая помощь будет оценена.Я довольно новичок в этом.

Ответ №1:

Используйте Layer.set_weights() .

Смотрите здесь https://www.tensorflow.org/api_docs/python/tf/keras/layers/Layer#set_weights