We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent d0b6bbb commit 03a677cCopy full SHA for 03a677c
tensorlayerx/backend/ops/load_backend.py
@@ -58,8 +58,8 @@
58
import mindspore.context as context
59
import os
60
os.environ['DEVICE_ID'] = '0'
61
- # context.set_context(mode=context.PYNATIVE_MODE, device_target='GPU'),
62
- context.set_context(mode=context.PYNATIVE_MODE, device_target='CPU'),
+ context.set_context(mode=context.PYNATIVE_MODE, device_target='GPU'),
+ # context.set_context(mode=context.PYNATIVE_MODE, device_target='CPU'),
63
# enable_task_sink=True, enable_loop_sink=True)
64
# context.set_context(mode=context.PYNATIVE_MODE, device_target='Ascend')
65
sys.stderr.write('Using MindSpore backend.\n')
0 commit comments