search

wenet-e2e / wekws

Production First and Production Ready End-to-End Keyword Spotting Toolkit
Apache License 2.0
444 stars 111 forks source link

Convert target to torch.int64 for cross_entropy #141

Closed wangtiance closed 1 year ago

wangtiance commented 1 year ago

On my machine the original code threw an error RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'

I followed https://github.com/wenet-e2e/wekws#installation to setup the environment so I'm curious if this error has ever occured to other people.