fix: ft_net_middle fails on batch size 1 due to torch.squeeze

[andrewwhitecdw]

Problem

fix: ft_net_middle fails on batch size 1 due to torch.squeeze

Fix

Replace:

        x = torch.cat((x0,x1),1)
        x = torch.squeeze(x)
        x = self.classifier(x)

with:

        x = torch.cat((x0,x1),1)
        x = x.view(x.size(0), -1)
        x = self.classifier(x)

Files changed

• reIDmodel.py