search

cornell-zhang / heterocl

HeteroCL: A Multi-Paradigm Programming Infrastructure for Software-Defined Heterogeneous Computing
https://cornell-zhang.github.io/heterocl/
Apache License 2.0
326 stars 92 forks source link

Create scheme from MXNet model #118

Closed faku99 closed 4 years ago

faku99 commented 5 years ago

Is there a way to create a scheme from an imported MXNet model without having to define it?

Something like:

sym, arg, aux = onnx_mxnet.import_model(ONNX_MODEL_PATH)

hcl.create_scheme(arg, sym)

or do I have to define the network architecture from scratch as in the LeNet inference tutorial?

seanlatias commented 5 years ago

For now you need to build the network manually as in the LeNet tutorial. However, we are working on enabling automatic import from other ML frameworks. This will be included in our next release.

faku99 commented 5 years ago

Is there a release date planned for the next release?

seanlatias commented 5 years ago

We plan to release it in fall.

seanlatias commented 4 years ago

142