ChainerX support

Open muupan opened this issue 7 years ago • 4 comments

We should use chainerx where possible for efficiency.

Dec 23 '18 10:12 muupan

~Links cannot be deepcopied after to_device('native'). We need to find a workaround or wait until it's fixed. https://github.com/chainer/chainer/issues/5916~ solved

Dec 23 '18 11:12 muupan

Async training requires https://github.com/chainer/chainer/issues/5931 to be fixed.

Dec 25 '18 09:12 muupan

def deepcopy_link(link):
    device = link.device
    link.to_device(np)
    new_link = copy.deepcopy(link)
    link.to_device(device)
    new_link.to_device(device)
    return new_link

This can be a workaround to deepcopy.

Dec 25 '18 09:12 muupan

Current ChainerX does not support advanced indexing, which prevents from applying it to CategoricalDQN and IQN. https://github.com/chainer/chainer/issues/5944

Dec 28 '18 09:12 muupan