Skip to content

Commit 03a677c

Browse files
authored
Update load_backend.py
1 parent d0b6bbb commit 03a677c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayerx/backend/ops/load_backend.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -58,8 +58,8 @@
5858
import mindspore.context as context
5959
import os
6060
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'),
61+
context.set_context(mode=context.PYNATIVE_MODE, device_target='GPU'),
62+
# context.set_context(mode=context.PYNATIVE_MODE, device_target='CPU'),
6363
# enable_task_sink=True, enable_loop_sink=True)
6464
# context.set_context(mode=context.PYNATIVE_MODE, device_target='Ascend')
6565
sys.stderr.write('Using MindSpore backend.\n')

0 commit comments

Comments
 (0)