Variable used in lines 13,105 and 147 in util.py is deprecated. Comme…

src/pytorch/functions.py

@@ -18,7 +18,7 @@ def __init__(self, k):
         '''
         super(linearUnified, self).__init__()
         self.k = k
-
+    @staticmethod
     def forward(self, x, w, b):
         '''
         forward propagation
@@ -36,7 +36,7 @@ def forward(self, x, w, b):
         self.add_buffer = x.new(x.size(0)).fill_(1)
         y.addr_(self.add_buffer, b)
         return y
-
+    @staticmethod
     def backward(self, dy):
         '''
         backprop with meprop

src/pytorch/modules.py

@@ -36,9 +36,9 @@ def reset_parameters(self):
 
     def forward(self, x):
         if self.unified:
-            return linearUnified(self.k)(x, self.w, self.b)
+            return linearUnified.apply(x, self.w, self.b)#add apply
         else:
-            return linear(self.k)(x, self.w, self.b)
+            return linear.apply(x, self.w, self.b)#add apply
 
     def __repr__(self):
         return '{} ({} -> {} <- {}{})'.format(self.__class__.__name__,

src/pytorch/util.py

@@ -10,7 +10,7 @@
 import torch.nn.functional as F
 import torch.optim as optim
 import torch.utils.data
-from torch.autograd import Variable
+from torch.autograd import Variable#deprecated- no longer supported
 
 from model import NetLayer
 
@@ -102,8 +102,7 @@ def _train(self, model, opt):
         utime = 0
         tloss = 0
         for bid, (data, target) in enumerate(self.trainloader):
-            data, target = Variable(data).cuda(), Variable(
-                target.view(-1)).cuda()
+            data, target = Variable(data).cuda(), Variable(target.view(-1)).cuda()#Deprecated- need to replace
             start = torch.cuda.Event(True)
             end = torch.cuda.Event(True)
 
@@ -145,8 +144,7 @@ def _evaluate(self, model, loader, name='test'):
         test_loss = 0
         correct = 0
         for data, target in loader:
-            data, target = Variable(
-                data, volatile=True).cuda(), Variable(target).cuda()
+            data, target = Variable(data, volatile=True).cuda(), Variable(target).cuda()# Deprecated- need to update
             output = model(data)
             test_loss += F.nll_loss(output, target).data[0]
             pred = output.data.max(1)[
@@ -204,7 +202,7 @@ def run(self, k=None, epoch=None):
             start = torch.cuda.Event(True)
             end = torch.cuda.Event(True)
             start.record()
-            loss, ft, bt, ut = self._train(model, opt)
+            loss, ft, bt, ut = self._train(model, opt)#model is trained here
             end.record()
             end.synchronize()
             ttime = start.elapsed_time(end)
@@ -215,7 +213,7 @@ def run(self, k=None, epoch=None):
             btime.append(bt)
             utime.append(ut)
             # predict
-            curacc = self._evaluate(model, self.devloader, 'dev')
+            curacc = self._evaluate(model, self.devloader, 'dev')#evaluation here
             if curacc > acc:
                 e = t
                 acc = curacc