Fixed wrt the source.

diff --git a/README.md b/README.md
index 4c219b7..9b4017c 100644 (file)
--- a/README.md
+++ b/README.md
@@ -46,7 +46,7 @@ criterion = nn.MSELoss()
 loss = criterion(output, target)
 
 agtree2dot.save_dot(loss,
-                    { input: 'input', loss: 'loss' },
+                    { input: 'input', target: 'target', loss: 'loss' },
                     open('./mlp.dot', 'w'))
 ```