【发布时间】:2016-11-14 21:50:34
【问题描述】:
我正在尝试使用 tensorflow 模型动物园中的 tensorflow 模型。由于我不熟悉 Bazel 编译过程,因此我对如何编译和使用这些模型感到困惑。有人知道它是怎么做的吗?
【问题讨论】:
标签: build tensorflow bazel
我正在尝试使用 tensorflow 模型动物园中的 tensorflow 模型。由于我不熟悉 Bazel 编译过程,因此我对如何编译和使用这些模型感到困惑。有人知道它是怎么做的吗?
【问题讨论】:
标签: build tensorflow bazel
首先使用blaze build构建一个target,运行blaze-bin目录下的target文件。例如。在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
【讨论】: