PyTorch requires padded sequence lengths to be on CPU

yannvgn / laserembeddings

LASER multilingual sentence embeddings as a pip package

BSD 3-Clause "New" or "Revised" License

224 stars 29 forks source link

PyTorch requires padded sequence lengths to be on CPU #32

Closed niklaskorz closed 3 years ago

niklaskorz commented 3 years ago

Related issues / commits from other projects:

https://github.com/mozilla/TTS/issues/567
https://github.com/pytorch/pytorch/issues/43227
https://github.com/joeynmt/joeynmt/pull/119/commits/ef1b0f081fedfcf83ce39cd2f27dcc33d3cbb2d1

yannvgn commented 3 years ago

Thank you for your PR, @niklaskorz

Can you please revert the changes in pyproject.toml? I want to keep the minimal versions of the dependencies as low as possible, and I want to keep supporting Python 3.6, just like in the original LASER project. I will also take care of bumping the version outside of this PR.