Open
Description
In the following function type of the middle two params are specified, but first two and last two arent. Also what device is compatible? It says CUDA in readme but when you use cuda tensors, the call to function compute_wer fails because it has .float() and cuda tensor doesnt have that casting. So are CPU tensors expected?
def levenshtein_distance(
hypotheses, # type: torch.Tensor
references, # type: torch.Tensor
hypothesis_lengths, # type: torch.IntTensor
references_lengths, # type: torch.IntTensor
blank, # type: torch.Tensor
separator # type: torch.Tensor
):
Metadata
Metadata
Assignees
Labels
No labels