Fix the CUDA/Hogwild test_seq2seq bug.
@@ -67,6 +67,8 @@ class TestSeq2Seq(unittest.TestCase):
"test ppl = {}\nLOG:\n{}".format(test['ppl'], stdout)
)
+
+class TestHogwildSeq2seq(unittest.TestCase):
def test_generation_multi(self):
"""This test uses a multi-turn task and multithreading."""
stdout, valid, test = _mock_train(
