【问题标题】:tensorflow models compilation张量流模型编译
【发布时间】:2016-11-14 21:50:34
【问题描述】:

我正在尝试使用 tensorflow 模型动物园中的 tensorflow 模型。由于我不熟悉 Bazel 编译过程,因此我对如何编译和使用这些模型感到困惑。有人知道它是怎么做的吗?

【问题讨论】:

    标签: build tensorflow bazel


    【解决方案1】:

    首先使用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
    

    【讨论】:

      猜你喜欢
      • 2018-11-25
      • 2019-06-18
      • 2017-11-22
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      • 1970-01-01
      相关资源
      最近更新 更多