2016-11-14 94 views
-1

我想尝试在tensorflow模型动物园中定位的tensorflow模型。由于我不熟悉Bazel编译过程,因此我不知道如何编译和使用这些模型。有谁知道它是如何完成的?tensorflow模型汇编

回答

1

首先使用blaze build建立一个目标并运行blaze-bin目录下的目标文件。例如。在inception model

# Build the model. Note that we need to make sure the TensorFlow is ready to 
# use before this as this command will not build TensorFlow. 
bazel build inception/imagenet_train 

# run it 
bazel-bin/inception/imagenet_train --num_gpus=1 --batch_size=32 --train_dir=/tmp/imagenet_train --data_dir=/tmp/imagenet_data