This repository was archived by the owner on Nov 17, 2023. It is now read-only.
Replies: 1 comment
-
@sandeep-krishnamurthy : Could you please tag this "Question", "Python" @ArneNx : You could post this question on "discuss.mxnet.io". You would receive a wider audience. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
https://github.com/apache/incubator-mxnet/blob/251ae71a20d8318ab20f4c19520f74b881fdf3ff/python/mxnet/module/bucketing_module.py#L353
I want to set the
grad_req
for all my parameters to'add'
by using specifying it in thebind()
call on my top-level module.The problem is that I use a
BucketingModule
. So when I callforward
, it automatically calls theswitch_bucket
method, which then callsbind
again without specifyinggrad_req
and therefore setting it to'write'
.Is this intended behavior? Can I somehow work around this while still using bucketing?
Beta Was this translation helpful? Give feedback.
All reactions