pytorch converter

neo-ai · Apr 2, 2019 · ac2146e · ac2146e
1 parent 5310e8d
commit ac2146e
Show file tree

Hide file tree

Showing 9 changed files with 1,249 additions and 0 deletions.
diff --git a/nnvm/python/nnvm/frontend/__init__.py b/nnvm/python/nnvm/frontend/__init__.py
@@ -7,3 +7,4 @@
 from .darknet import from_darknet
 from .tensorflow import from_tensorflow
 from .caffe2 import from_caffe2
+from .pytorch import from_pytorch
diff --git a/nnvm/python/nnvm/frontend/pytorch/__init__.py b/nnvm/python/nnvm/frontend/pytorch/__init__.py
@@ -0,0 +1,2 @@
+r'''PyTorch->NNVM converter'''
+from .converter import from_pytorch