From: Francois Fleuret Date: Wed, 8 Mar 2017 10:47:16 +0000 (+0100) Subject: Fixed wrt the source. X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=agtree2dot.git;a=commitdiff_plain;h=8a79b8841aea1fee0d1749a0650edfed196d5086 Fixed wrt the source. --- 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')) ```