Обработка тензорного потока при обучении пользовательских объектов - PullRequest
Новая
       9

Обработка тензорного потока при обучении пользовательских объектов

0 голосов
/ 06 мая 2018

Я хочу обучить множество изображений в соответствии с моими исследованиями. следование этому уроку приводит к ошибке. это учебник, которому я следую. Учебник

Я внедряю этот проект в Python 3.6, последнюю версию tenorflow и его версию процессора, в то время как я запускаю выполнение, чтобы обучить модель в соответствии с руководством, она выдает ошибку, подобную этой. 

    Instructions for updating:
    Please switch to tf.train.create_global_step
    INFO:tensorflow:Scale of 0 disables regularizer.
    INFO:tensorflow:Scale of 0 disables regularizer.
    INFO:tensorflow:depth of additional conv before box predictor: 0
    WARNING:tensorflow:From C:\tensorflow1\models\research\object_detection\core\box_predictor.py:403: calling reduce_mean (from tensorflow.python.ops.math_ops) with keep_dims is deprecated and will be removed in a future version.
    Instructions for updating:
    keep_dims is deprecated, use keepdims instead
    INFO:tensorflow:Scale of 0 disables regularizer.
    WARNING:tensorflow:From C:\tensorflow1\models\research\object_detection\core\losses.py:317: softmax_cross_entropy_with_logits (from tensorflow.python.ops.nn_ops) is deprecated and will be removed in a future version.
    Instructions for updating:

    Future major versions of TensorFlow will allow gradients to flow
    into the labels input on backprop by default.

    See @{tf.nn.softmax_cross_entropy_with_logits_v2}.

    Traceback (most recent call last):
      File "train.py", line 167, in <module>
        tf.app.run()
      File "C:\Anaconda\envs\tensorflow\lib\site-packages\tensorflow\python\platform\app.py", line 126, in run
        _sys.exit(main(argv))
      File "train.py", line 163, in main
        worker_job_name, is_chief, FLAGS.train_dir)
      File "C:\tensorflow1\models\research\object_detection\trainer.py", line 284, in train
        train_config.optimizer)
      File "C:\tensorflow1\models\research\object_detection\builders\optimizer_builder.py", line 50, in build
        learning_rate = _create_learning_rate(config.learning_rate)
      File "C:\tensorflow1\models\research\object_detection\builders\optimizer_builder.py", line 109, in _create_learning_rate
        learning_rate_sequence, config.warmup)
      File "C:\tensorflow1\models\research\object_detection\utils\learning_schedules.py", line 156, in manual_stepping
        raise ValueError('First step cannot be zero.')
    ValueError: First step cannot be zero.

Почему я получаю такую ​​ошибку? пожалуйста, помогите мне исправить это.

1 Ответ

0 голосов
/ 26 июля 2018

Я нашел исправление для этого здесь:

https://github.com/tensorflow/models/issues/3794

В основном, извлеките этот код из вашего обучающего файла / fast_rcnn_inception_v2_pets.config (или как вы его назвали): 

schedule {
   step: 0
   learning_rate: .0001
}

Если вы ищете именно этот текст, вы можете его не найти, в моем коде это было learning_rate: .0002. В любом случае, надеюсь, это поможет.

...