X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=agtree2dot.git;a=blobdiff_plain;f=README.md;fp=README.md;h=9b4017c9c3e1773fab8578d08b3cba5e4c4214c9;hp=4c219b713b8e2f3199a66ead5e6dd46082153076;hb=8a79b8841aea1fee0d1749a0650edfed196d5086;hpb=c41d87166cb2a6dc04def94da1f54d44030046ea

diff --git a/README.md b/README.md
index 4c219b7..9b4017c 100644
--- 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'))
 ```