Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

ChainerX support #373

Open
muupan opened this issue Dec 23, 2018 · 4 comments
Open

ChainerX support #373

muupan opened this issue Dec 23, 2018 · 4 comments

Comments

@muupan
Copy link
Member

muupan commented Dec 23, 2018

We should use chainerx where possible for efficiency.

@muupan
Copy link
Member Author

muupan commented Dec 23, 2018

Links cannot be deepcopied after to_device('native'). We need to find a workaround or wait until it's fixed. chainer/chainer#5916 solved

@muupan
Copy link
Member Author

muupan commented Dec 25, 2018

Async training requires chainer/chainer#5931 to be fixed.

@muupan
Copy link
Member Author

muupan commented Dec 25, 2018

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.

@muupan
Copy link
Member Author

muupan commented Dec 28, 2018

Current ChainerX does not support advanced indexing, which prevents from applying it to CategoricalDQN and IQN. chainer/chainer#5944

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

1 participant