make it clear in the example that residual not included

LLYX · Jul 27, 2020 · ae05f37 · ae05f37
1 parent 321a620
commit ae05f37
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -14,7 +14,7 @@ $ pip install conformer
 
 ```python
 import torch
-from conformer.conformer import ConformerConvModule
+from conformer import ConformerConvModule
 
 layer = ConformerConvModule(
     dim = 512,
@@ -25,7 +25,7 @@ layer = ConformerConvModule(
 )
 
 x = torch.randn(1, 1024, 512)
-layer(x) # (1, 1024, 512)
+x = layer(x) + x
 ```
 
 ## Citations