A strange mistake in triplet_loss.py

[mohuang1996]

hi!
I run your code but reported a strange error:File "./code/triplet_loss.py", line 62, in hard_example_mining dist_ap, relative_p_inds = torch.max(dist_mat[is_pos].contiguous().view(N, -1), 1, keepdim=True)
RuntimeError: shape '[64, -1]' is invalid for input of size 288
I think it for a long time, but I don't know how to fix it, can anyone help me？