Skip to content
Snippets Groups Projects
Commit f32754fc authored by i-robot's avatar i-robot Committed by Gitee
Browse files

!299 fix textrcnn-gru gpu bug

Merge pull request !299 from chenweitao_295/textrcnn_gpu_issues
parents de73d159 7203f6ac
No related branches found
No related tags found
No related merge requests found
......@@ -52,7 +52,7 @@ def run_train():
device_id = get_device_id()
context.set_context(device_id=device_id)
if cfg.device_target == "GPU":
if cfg.device_target == "GPU" and cfg.cell == "lstm":
context.set_context(enable_graph_kernel=True)
if cfg.preprocess == 'true':
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment