ZiyueHuang commented on a change in pull request #8539: expose group2ctx to
module
URL: https://github.com/apache/incubator-mxnet/pull/8539#discussion_r149581557
##########
File path: python/mxnet/module/module.py
##########
@@ -59,10 +59,12 @@ class Module(BaseModule):
state_names : list of str
states are similar to data and label, but not provided by data
iterator.
Instead they are initialized to 0 and can be set by `set_states()`.
+ group2ctxs : list of dict of str to context
+ Default is `None`. Mapping the `ctx_group` attribute to the context
assignment.
"""
def __init__(self, symbol, data_names=('data',),
label_names=('softmax_label',),
logger=logging, context=ctx.cpu(), work_load_list=None,
- fixed_param_names=None, state_names=None):
+ fixed_param_names=None, state_names=None, group2ctxs=None):
Review comment:
I think it's better for `group2ctxs` to be in the same place with `contexts`.
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
