Skip to content
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

fix two potential bugs #1

Open
wants to merge 1 commit into
base: master
Choose a base branch
from
Open

fix two potential bugs #1

wants to merge 1 commit into from

Commits on Aug 24, 2021

  1. fix two bugs 

    Two changes:
1. `mask = mask.to(output[0].device).unsqueeze(-1)` -> `mask = mask.to(output[0]).unsqueeze(-1)`, to assure mask has the same device and data type as output[0]
2. add outbound error check. If all heads are sampled to zero, then` a tensor / count_ones` will raise "Division by 0 " Error.
    @frankang
    frankang authored Aug 24, 2021
    Configuration menu
    Copy the full SHA
    07c9912 View commit details
    Browse the repository at this point in the history