From 518160de26086ae87643744e765440e4d92ebb42 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Sun, 20 May 2018 23:00:01 +0200 Subject: [PATCH] Fixed a bug with 0.4.0 + typo. --- README.md | 2 +- mlp.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) 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) -- 2.20.1