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

Before Change

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

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

After Change

            // 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

Instances