search

IBM / pytorch-seq2seq

An open source framework for seq2seq models in PyTorch.
https://ibm.github.io/pytorch-seq2seq/public/index.html
Apache License 2.0
1.5k stars 376 forks source link

Integration test expected cuda bug resolved #161

Closed pskrunner14 closed 6 years ago

pskrunner14 commented 6 years ago

Made changes to test due to deprecation warnings. Added cuda assignments to variables in TopKDecoder.py. Fixes #160.