如何还原Inception V3的权重并仅重新训练最后一层?

时间:2018-07-10 19:04:03

标签: python tensorflow

我已成功遵循本教程(https://codelabs.developers.google.com/codelabs/tensorflow-for-poets/#1)。我已经对自己的数据进行了训练,并且行得通。但是,现在我想在我花了一天时间训练的最后一层上继续训练。但是,当我简单地重新运行命令

python -m scripts.retrain \
  --bottleneck_dir=tf_files/bottlenecks \
  --how_many_training_steps=500 \
  --model_dir=tf_files/models/ \
  --summaries_dir=tf_files/training_summaries/"${ARCHITECTURE}" \
  --output_graph=tf_files/retrained_graph.pb \
  --output_labels=tf_files/retrained_labels.txt \
  --architecture="${ARCHITECTURE}" \
  --image_dir=tf_files/flower_photos

培训从0开始。

我想这样做,因为我希望随着学习的持续进行调整学习率(以及其他可能的价值)

特别有趣的是retrain.py文件(https://github.com/googlecodelabs/tensorflow-for-poets-2/tree/master/scripts)。我还尝试将一些代码从label_image.py复制到retrain.py,但没有成功。

0 个答案:

没有答案