Hi pytorch community. I am having problems using torch.nn.functional.ctc_loss()
I have the following tensors:
prob_txt_pred: torch.Size([225, 51, 54]) target_words: torch.Size([51, 30]) length_input: torch.Size() target_len_words: torch.Size()
I am giving them to the ctc_loss function in the following way:
nf.ctc_loss(prob_txt_pred, target_words, length_input, target_len_words)
and I am getting an error that says that dimension 1 of the tensor: ‘targets’ (in this case tensor_words) is expected to be at least 225.
The full error says:
RuntimeError: Expected tensor to have size at least 225 at dimension 1, but got size 51 for argument #2 ‘targets’ (while checking arguments for ctc_loss_gpu)
Any idea why?
thanks for your time!