如何使用我自己的张量流自定义数据训练新模型

Question

请原谅我，我对Tensorflow很新，所以请尽量详细说明。非常感谢!!!

我现在的问题是我已经用一些参数

运行了这个脚本

https://github.com/tensorflow/models/blob/master/inception/inception/data/build_image_data.py

# python build_image_data.py --train_directory /Input --validation_directory /validation --output_directory /output --labels_file /labels_file

构建图像数据并获得一些输出，例如 train-00000-of-00002，train-00001-of-00002，validation-00000-of-00002，validation-00001-of-00002

之后，如何在张量流上训练具有上述自定义数据的新模型？

非常感谢！

Answer 1

我自己解决了这个问题，谢谢大家。

http://blog.twman.org/2016/06/tensorflow.html

bazel build inception/build_image_data

bazel-bin/inception/build_image_data   --train_directory="${TRAIN_DIR}"  \
--validation_directory="${VALIDATION_DIR}"   \
--output_directory="${OUTPUT_DIRECTORY}"   \
--labels_file="${LABELS_FILE}"   \
--train_shards=128   \
--validation_shards=24   \
--num_threads=8