1

您能否告诉我在训练/测试 SyntaxNet 时如何指定要使用的实例?

在其他 tensorflow 模型中,我们可以通过编辑 Python 代码轻松更改配置:

例如)tf.device('/cpu:0')=> tf.device('/gpu:0')

我可以通过运行运行 parsey mcparseface 模型,demo.sh然后我按照符号链接查找设备配置。

demo.sh也许我错过了但是我在和parser_eval.py中找不到 gpu 配置 python 代码context.proto

device当我在tensorflow/models中使用查询“”进行搜索时,我可以看到几个 C 文件,例如syntaxnet/syntaxnet/unpack_sparse_features.cc包含行using tensorflow::DEVICE_CPU;

那么.. 更改这些文件中的 C 代码是更改 SyntaxNet 设备配置的唯一方法吗?

我希望有一种更简单的方法来更改 Python 级别的设置。

提前致谢。

4

1 回答 1

0

有关在 GPU 上运行语法网络的说明,您可以参考此页面:https ://github.com/tensorflow/models/issues/248

Tensorflow 会自动将包括 GPU 在内的设备分配给操作:https ://www.tensorflow.org/versions/r0.11/how_tos/using_gpu/index.html 。您还可以在构建图形时手动指定设备。

于 2016-11-15T07:11:20.150 回答