Merge pull request #3 from kengz/expose

expose to_namedtuple
kengz · Jun 15, 2020 · 895a42e · 895a42e
2 parents 898b216 + 014d4b6
commit 895a42e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -358,7 +358,7 @@ batch_size = 16
 dag_in_shape = arc['dag_in_shape']
 data = {'image': torch.rand([batch_size, *dag_in_shape['image']]), 'vector': torch.rand([batch_size, *dag_in_shape['vector']])}
 # convert from a dict of Tensors into a TensorTuple - a namedtuple
-xs = torcharc.net_util.to_namedtuple(data)
+xs = torcharc.to_namedtuple(data)
 # returns TensorTuple if output is multi-model, Tensor otherwise
 ys = model(xs)
 ```

diff --git a/setup.py b/setup.py
@@ -35,7 +35,7 @@ def run_tests(self):
 
 setup(
     name='torcharc',
-    version='0.0.3',
+    version='0.0.4',
     description='Build PyTorch networks by specifying architectures.',
     long_description='https://github.com/kengz/torcharc',
     keywords='torcharc',

diff --git a/torcharc/__init__.py b/torcharc/__init__.py
@@ -1,4 +1,5 @@
 from torcharc import module_builder
+from torcharc.net_util import to_namedtuple
 from torcharc.module import dag
 from torch import nn