Skip to content

Conversation

@astariul
Copy link

For the baseline function (in train_abstractive.py), device ID is given as a constant (-1, because baseline don't need GPU).

This constant was a str, giving TypeError when run, because it is compared to a int in build_trainer.

Modified to int, which fix the problem.

For baseline function, device ID is given as a constant (-1, because baseline don't need GPU). This constant was a `str`, giving TypeError. Modified to int.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant