7fffa98b9166a03b4a53fb40202d97b09e8e9036,models.py,YOLOLayer,forward,#YOLOLayer#,94

Before Change

                                                                                                        self.seen)


            nProposals = int((conf > 0.25).sum().data[0])

            tx    = Variable(tx.type(FloatTensor), requires_grad=False)
            ty    = Variable(ty.type(FloatTensor), requires_grad=False)

After Change

        FloatTensor = torch.cuda.FloatTensor if x.is_cuda else torch.FloatTensor
        LongTensor = torch.cuda.LongTensor if x.is_cuda else torch.LongTensor

        if x.is_cuda:
            self.mse_loss = self.mse_loss.cuda()
            self.ce_loss = self.ce_loss.cuda()

        prediction = x.view(bs, self.bbox_attrs * self.num_anchors, g_dim * g_dim)
        prediction = prediction.transpose(1, 2).contiguous()
        prediction = prediction.view(bs, g_dim * g_dim * self.num_anchors, self.bbox_attrs)

Instances