Open jixiaw opened 4 years ago
In the forward function of class PNet: "a = F.softmax(a, dim=-1)", why the dim is -1? Isn't it 1.
In the forward function of class PNet: "a = F.softmax(a, dim=-1)", why the dim is -1? Isn't it 1.