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

Faster forward pass + tests on GPU #123

Open
wants to merge 3 commits into
base: master
Choose a base branch
from
Open

Faster forward pass + tests on GPU #123

wants to merge 3 commits into from

Commits on Oct 30, 2024

  1. Now calculating the _compute_score method without iterating over the … 

    …sequence length with a python loop. The result is the same, but faster
    @Academich
    Academich committed Oct 30, 2024
    Configuration menu
    Copy the full SHA
    39c8f43 View commit details
    Browse the repository at this point in the history

  2. Changed the mask type from ByteTensor to BoolTensor. Pytorch throws w… 

    …arnings that byte tensor masks are deprecated
    @Academich
    Academich committed Oct 30, 2024
    Configuration menu
    Copy the full SHA
    fed5f80 View commit details
    Browse the repository at this point in the history

  3. Added a test that checks if the forward pass of the CRF module works … 

    …on a GPU
    @Academich
    Academich committed Oct 30, 2024
    Configuration menu
    Copy the full SHA
    966af5a View commit details
    Browse the repository at this point in the history