search

awslabs / raf

Apache License 2.0
140 stars 20 forks source link

How can I load an ONNX model into raf? #118

Closed ikaroso closed 2 years ago

ikaroso commented 2 years ago

Hi, it is said in the doc that raf supports ONNX, but I cant find how to load an ONNX model in raf.

hgt312 commented 2 years ago

Some description in wiki is out of date. RAF does not have a ONNX frontend now. To load a ONNX model, use TVM's ONNX frontend with RAF pass FromRelay may works. FYI: https://github.com/awslabs/raf/blob/main/python/raf/frontend/pytorch.py