session.run([операция,потеря],feedDictionary) не работает
Добрый день,
Я новичок в тензорном потоке. Я пытаюсь запустить этот код github , который сегментирует изображения niifi.
Я подозреваю, что tensorflow не запускает следующий фрагмент:
_, loss_value = sess.run([train_op, loss], feed_dict=feed_dict)
этот код генерирует параметр train_op
train_op = model.training_step(loss, exp_config.optimizer_handle, learning_rate_pl)
def training_step(loss, optimizer_handle, learning_rate, **kwargs):
'''
Creates the optimisation operation which is executed in each training iteration of the network
:param loss: The loss to be minimised
:param optimizer_handle: A handle to one of the tf optimisers
:param learning_rate: Learning rate
:param momentum: Optionally, you can also pass a momentum term to the optimiser.
:return: The training operation
'''
if 'momentum' in kwargs:
momentum = kwargs.get('momentum')
optimizer = optimizer_handle(learning_rate=learning_rate, momentum=momentum)
else:
optimizer = optimizer_handle(learning_rate=learning_rate)
# The with statement is needed to make sure the tf contrib version of batch norm properly performs its updates
update_ops = tf.get_collection(tf.GraphKeys.UPDATE_OPS)
with tf.control_dependencies(update_ops):
train_op = optimizer.minimize(loss)
return train_op
после печати train_op у меня есть это:
имя: "Адам"
Оп: "Нет Оп"
ввод: "^conv1_1_bn/cond_2/Объединить"
ввод: "^conv1_1_bn/cond_3/Объединить"
ввод: "^conv1_2_bn/cond_2/Объединить"
...
... (больше текста)
...
ввод: "^Adam/update_conv8_2_bn/beta/ApplyAdam"
ввод: "^Адам/update_pred/W/ApplyAdam"
ввод: "^Адам/update_pred_bn/gamma/ApplyAdam"
ввод: "^Адам/update_pred_bn/beta/ApplyAdam"
ввод: "^Адам/Назначить"
ввод: "^Адам/Назначить_1"
Я не знаю, что здесь происходит, но я проверил, что сеанс запущен
print(sess._opened)
Я использую тензорный поток GPU ==1.12
как вы думаете, что я делаю неправильно? Благодарность