如何恢复Inception V3的权重并只重新训练最后一层?

2024-09-27 09:26:08 发布

您现在位置:Python中文网/ 问答频道 /正文

我成功地遵循了本教程(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开始。在

我想这样做,因为我希望随着学习的继续调整学习率(以及潜在的其他值)

尤其是再培训.py文件(https://github.com/googlecodelabs/tensorflow-for-poets-2/tree/master/scripts)很有趣。我还试着从标签上复制一些代码_图像.py进入再培训.py但是没有成功。在


Tags: pyhttpscomforoutputtftensorflowdir

热门问题