f466871be6ee80533c997cf6c958aa41a697936f,trainer.py,Trainer,train_one_epoch,#Trainer#,137

Before Change

            img, target = Variable(img), Variable(target)

            // initialize location and hidden state vectors
            h_t = torch.zeros(self.batch_size, self.hidden_size)
            l_t = torch.Tensor(self.batch_size, 2).uniform_(-1, 1)
            h_t, l_t = Variable(h_t), Variable(l_t)

            for t in range(self.num_glimpses - 1):
                // forward pass through model
                h_t, l_t = self.model(img, l_t, h_t)

                // bookeeping for later plotting
                self.locs.append(l_t)

            // last iteration
            probas = self.model(img, l_t, h_t, last=True)

            // to be continued

After Change

        tic = time.time()
        for i, (x, y) in enumerate(self.train_loader):
            x, y = Variable(x), Variable(y)

            self.clear()

Instances