Как вы передаете объект сессии в TensorFlow v2? - PullRequest
0 голосов
/ 20 апреля 2020

У меня есть функция change_weight(), которая изменяет веса в любой модели. Эта функция находится в другом файле python.

Так что, если у меня есть простая нейронная сеть, которая классифицирует изображения MNIST, я проверяю точность до и после вызова этой функции и вижу, что она работает. Это было легко сделать в TensorFlow v1, так как мне просто нужно было передать объект Session sess в вызов функции, и я мог получить вес этого сеанса в другом файле.

При стремительном выполнении в TensorFlow v2, как мне это сделать? У меня больше нет объекта Session. Что я прохожу?

1 Ответ

0 голосов
/ 20 апреля 2020

Я смог сделать это, передав вместо этого объект Model и получив веса на model.trainable_variables в другой функции.

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