From: Francois Fleuret Date: Sun, 20 May 2018 21:00:01 +0000 (+0200) Subject: Fixed a bug with 0.4.0 + typo. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=518160de26086ae87643744e765440e4d92ebb42;p=agtree2dot.git Fixed a bug with 0.4.0 + typo. --- diff --git a/README.md b/README.md index 452aa9c..f290eca 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,7 @@ This package provides a function that generates a [dot file](https://en.wikipedia.org/wiki/DOT_(graph_description_language)) -from a [pytorch](http://pytorch.org) autograd graph. +from a [PyTorch](http://pytorch.org) autograd graph. # Usage # diff --git a/mlp.py b/mlp.py index 7ddb894..21d166b 100755 --- a/mlp.py +++ b/mlp.py @@ -22,7 +22,6 @@ 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 @@ -40,8 +39,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)