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

как вы думаете, что я делаю неправильно? Благодарность

0 ответов

Другие вопросы по тегам