New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Set device of start class token in autoregressive log_prob #43

Merged

srush merged 2 commits into harvardnlp:master from justinchiu:master

Dec 20, 2019

Contributor

justinchiu commented Dec 20, 2019

There was a device mismatch when using Autoregressive.log_prob on gpu, the start_class tensor was constructed on the default device.


          set device of bos in autoregressive log_prob

Collaborator

srush commented Dec 20, 2019

./autoregressive.py:91:21: E131 continuation line unaligned for hanging indent

(run black.py before submitting)


          fix lint

ebaefdb

srush merged commit 834c849 into harvardnlp:master

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet