From 518160de26086ae87643744e765440e4d92ebb42 Mon Sep 17 00:00:00 2001
From: Francois Fleuret <francois@fleuret.org>
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.39.5