8000
We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c5f0551 commit e050b85Copy full SHA for e050b85
tests/chainermn_tests/datasets_tests/test_mnist.py
@@ -46,7 +46,6 @@ def check_mnist(use_gpu, use_chx, display_log=True):
46
chainer.cuda.get_device_from_id(comm.intra_rank).use()
47
48
device = get_device(comm.intra_rank if use_gpu else None, use_chx)
49
- device.use()
50
model.to_device(device)
51
optimizer = chainermn.create_multi_node_optimizer(
52
chainer.optimizers.Adam(), comm)
0 commit comments