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