Updated with DAG:setLabel()

[dagnn.git] / README.md

diff --git a/README.md b/README.md
index 4c356dc..c641e03 100644 (file)
--- a/README.md
+++ b/README.md
@@ -76,7 +76,7 @@ respectively.
 
 Create a new empty DAG, which inherits from nn.Container.
 
-### nn.DAG:connect(module1, module2 [, module3, [...]]]) ###
+### nn.DAG:connect(module1, module2 [, module3, [...]]) ###
 
 Add new nodes corresponding to the modules passed as arguments if they
 have not been already added in a previous call. Add edges between
@@ -115,3 +115,7 @@ a pdf file with
 ```
 dot graph.dot -T pdf -o graph.pdf
 ```
+
+### nn.DAG:setLabel(module, name) ###
+
+Add a label to the given module, that will be used for DAG:print() and DAG:saveDot()