【发布时间】:2016-11-15 06:18:34
【问题描述】:
您能否告诉我在训练/测试 SyntaxNet 时如何指定要使用的实例?
在其他 tensorflow 模型中,我们可以通过编辑 Python 代码轻松更改配置:
例如)tf.device('/cpu:0') => tf.device('/gpu:0')。
我可以通过运行 demo.sh 来运行 parsey mcparseface 模型,然后我返回符号链接来查找设备配置。
也许我错了但是我在demo.sh、parser_eval.py和context.proto中找不到gpu配置python代码。
当我在 tensorflow/models 中使用查询“device”进行搜索时,我可以看到几个 C 文件,例如 syntaxnet/syntaxnet/unpack_sparse_features.cc 包含行 using tensorflow::DEVICE_CPU;
那么.. 更改这些文件中的 C 代码是更改 SyntaxNet 设备配置的唯一方法吗?
我希望有更简单的方法来更改 Python 级别的设置。
提前致谢。
【问题讨论】:
标签: python nlp tensorflow gpu syntaxnet