From 8a79b8841aea1fee0d1749a0650edfed196d5086 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Wed, 8 Mar 2017 11:47:16 +0100 Subject: [PATCH] Fixed wrt the source. --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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')) ``` -- 2.20.1