Error in Lovasz loss

[seefun]

pytorch-toolbelt/pytorch_toolbelt/losses/lovasz.py

Line 150 in f3acfca

probas = torch.movedim(probas, 0, -1) # [B, C, Di, Dj, Dk...] -> [B, C, Di...Dk, C]

Should be:

probas = torch.movedim(probas, 1, -1)  # [B, C, Di, Dj, ...] -> [B, Di, Dj, ..., C]