从检查点还原后重新初始化参数-Tensorflow Object Detection API

时间:2019-12-13 13:01:03

标签: python tensorflow machine-learning object-detection object-detection-api

我正在尝试在GPU上使用Tensorflow(1.14)Object Detection API训练对象检测器。 因此,我正在使用tf模型动物园中的预训练模型。

培训配置为:

fine_tune_checkpoint: "/home/daniel/Desktop/venv/models_latest/research/object_detection/training/experiments/faster_rcnn_3/model.ckpt"
  from_detection_checkpoint: true
  num_steps: 1
  load_all_detection_checkpoint_vars: true
  fine_tune_checkpoint_type: "detection"

一旦我训练一张图像一步,这就是输出:

I1213 13:22:52.498878 140298995029824 saver.py:1280] Restoring parameters from /home/daniel/Desktop/venv/models_latest/research/object_detection/training/experiments/faster_rcnn_pets_3/model.ckpt
I1213 13:22:53.028416 140298995029824 session_manager.py:500] Running local_init_op.
I1213 13:22:53.295333 140298995029824 session_manager.py:502] Done running local_init_op.
Summary of weights: -7745.5962343811989
I1213 13:22:58.046248 140298995029824 learning.py:779] Starting Session.
I1213 13:22:58.180775 140292514621184 supervisor.py:1117] Saving checkpoint to path training/experiments/faster_rcnn_pets_3/training/model.ckpt
I1213 13:22:58.185102 140298995029824 learning.py:793] Starting Queues.
I1213 13:22:59.394556 140292506228480 supervisor.py:1099] global_step/sec: 0
I1213 13:23:04.981492 140291348625152 supervisor.py:1050] Recording summary at step 0.
I1213 13:23:09.964757 140298995029824 learning.py:519] global step 1: loss = 6.9496 (11.652 sec/step)
I1213 13:23:10.085421 140298995029824 learning.py:802] Stopping Training.
I1213 13:23:10.085600 140298995029824 learning.py:810] Finished training! Saving model to disk.
Summary of weights: -7745.7172140061102

似乎从检查点恢复了参数,但是我猜“运行中的local_init_op”以某种方式覆盖了它们?

此外,负大权重是什么意思?

0 个答案:

没有答案