search

emanjavacas / pie

A fully-fledge PyTorch package for Morphological Analysis, tailored to morphologically rich and historical languages.
MIT License
22 stars 10 forks source link

Fix a bug where pack_padded_sequence would take a cuda tensor instead of a CPU one #81

Closed PonteIneptique closed 3 years ago

PonteIneptique commented 3 years ago

Of course there had to be a regression / change I did not catch in my different tests. So here you go: pack_added_sequence requires a CPU tensor no instead of a cuda one.