While training viti I am getting thsi error #31
Description
Predictions and targets are expected to have the same shape, but got torch.Size([8, 400]) and torch.Size([8]).
{{ message }}
Predictions and targets are expected to have the same shape, but got torch.Size([8, 400]) and torch.Size([8]).