在TensorFlow从Scratch训练开始

时间:2016-08-30 12:41:17

标签: python machine-learning tensorflow

我想训练初始模型,如tensorflow github-tutorial中所示。 除了我想使用自制的TFRecord文件数据集。

bazel build inception/imagenet_train
bazel-bin/inception/imagenet_train --num_gpus=1 --batch_size=32 --train_dir=/tmp/imagenet_train --data_dir=/tmp/imagenet_data

我使用自己的TFRecord文件将数据目录更改为文件夹。 现在我想知道我是否真的从头开始训练,或者这是否像“重新训练最后一层 - 教程”一样

1 个答案:

答案 0 :(得分:0)

是的,您正在从头开始培训:请参阅code