Получение списка обучаемых переменных для использования tf.Optimizers.minimize - PullRequest
0 голосов
/ 21 апреля 2020

Я пытаюсь обучить CNN, реализованный «вручную», без использования функций keras.models. На данный момент я реализовал функции для создания слоев, функцию потерь и функцию вывода стеков слоев. До этого момента все работает.

Для учебной операции я вызываю Optimizer.minimize (), чтобы обновить градиенты, но для этого необходимо передать «var_list».

Я использовал следующий код для получения списка:

var_names2 = []    
for var in tf.compat.v1.trainable_variables():    
     var_names2.append(var.name)

Однако я получаю сообщение об ошибке: 'tenorflow. python .framework.ops.EagerTensor' объект не может быть вызван ' когда я пытаюсь выполнить тренировочную эпоху.

Не могли бы вы помочь мне найти этот список переменных? Спасибо!

...