Skip to content
This repository has been archived by the owner on Jul 2, 2021. It is now read-only.

Fix ConstantStubLink for chainer==6.0.0b2 #774

Closed
wants to merge 1 commit into from

Conversation

ktns
Copy link
Contributor

@ktns ktns commented Feb 10, 2019

In chainer==6.0.0b2, chainer.Chain.to_gpu/to_cpu no longer invokes to_gpu/to_cpu of child links, but directly enumerates parameters and persistent values and convert arrays accordingly.
This change affects ConstantStubLink because its overridden to_gpu/to_cpu is ignored by parent Chains.

This PR addresses the problem using add_persistent to register output values of ConstantStubLink.

@Hakuyume
Copy link
Member

According to chainer team, the APIs of to_cpu/to_gpu/to_device in 6.0.0b2 is not fixed and I asked to support our situation. I think we should wait for the decision.

@Hakuyume
Copy link
Member

A hotfix is added in #823

@Hakuyume Hakuyume closed this Mar 20, 2019
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants