AoANet
AoANet copied to clipboard
Fixed training errors
These edits fix these errors while training:
- 'lengths' should be a 1D CPU int64 tensor, but got 1D cuda:0 Long Tensro
- TypeError: can only concatenate str(not "float") to str