X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=agtree2dot.git;a=blobdiff_plain;f=mlp.py;fp=mlp.py;h=21d166bba9447a87315776cd741ad6d6f591e841;hp=7ddb894dd7248a54861dad3e9ef9af585a2c6372;hb=518160de26086ae87643744e765440e4d92ebb42;hpb=546094cd63776f6d2a5d6ce1a0fa935a7b9dc956

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)