2016-11-14 101 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