Add missing __init__.py code for src/python dir

insop · Sep 19, 2020 · e7d77e1 · e7d77e1
1 parent a06f9ed
commit e7d77e1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/python/__init__.py b/src/python/__init__.py
@@ -0,0 +1 @@
+from .transformer_simple import SelfAttention, MultiHeadAttention, TransformerBlock