From: Francois Fleuret Date: Sun, 20 May 2018 21:02:26 +0000 (+0200) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=765afb07c2781c92805461eb937d611fe03ca9bf;p=agtree2dot.git Update. --- diff --git a/README.md b/README.md index f290eca..6980672 100644 --- a/README.md +++ b/README.md @@ -20,12 +20,9 @@ resulting graph. A typical use is provided in [mlp.py](https://fleuret.org/git-extract/agtree2dot/mlp.py): ```python -import subprocess - from torch import nn from torch.nn import functional as fn from torch import Tensor -from torch.autograd import Variable from torch.nn import Module import agtree2dot @@ -43,8 +40,8 @@ class MLP(Module): return x mlp = MLP(10, 20, 1) -input = Variable(Tensor(100, 10).normal_()) -target = Variable(Tensor(100).normal_()) +input = Tensor(100, 10).normal_() +target = Tensor(100, 1).normal_() output = mlp(input) criterion = nn.MSELoss() loss = criterion(output, target) @@ -64,7 +61,8 @@ agtree2dot.save_dot(loss, print('Generated mlp.dot') try: - subprocess.check_call(["dot", "mlp.dot", "-Lg", "-T", "pdf", "-o", "mlp.pdf" ]) + subprocess.check_call(['dot', 'mlp.dot', '-Lg', '-T', 'pdf', '-o', 'mlp.pdf' ]) + except subprocess.CalledProcessError: print('Calling the dot command failed. Is Graphviz installed?') sys.exit(1) diff --git a/mlp.pdf b/mlp.pdf index a72a3b7..0a7b858 100644 Binary files a/mlp.pdf and b/mlp.pdf differ