Ray Zhang created MXNET-652:
-------------------------------
Summary: Allow for multi-gpu load from onnx into gluon
Key: MXNET-652
URL: https://issues.apache.org/jira/browse/MXNET-652
Project: Apache MXNet
Issue Type: Improvement
Reporter: Ray Zhang
Currently, Airbnb is developing applications that use mxnet(particularly
`gluon`). We use the `mxnet.contrib.onnx` package, which supports importing to
gluon from ONNX format. As this is an extremely important part of our pipeline,
we want to add support to multi-gpu imports, which is currently not supported.
This is actually a very simple change, because `_load_init()` already supports,
directly, a `Context` object or an iterable of them. However, the current
implementation for `import_to_gluon` suggests a `str` of `'CPU'` or `'GPU'`.
This is not only different from other parts of the mxnet codebase, but also
does not allow us to specify which gpu's we want to load the model into.
--
This message was sent by Atlassian JIRA
(v7.6.3#76005)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
