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

Keep model_avg on cpu #348

Merged
merged 10 commits into from
May 7, 2022
Merged

Keep model_avg on cpu #348

merged 10 commits into from
May 7, 2022

Commits on Apr 29, 2022

  1. Merge remote-tracking branch 'k2-fsa/master'

    @yaozengwei
    yaozengwei committed Apr 29, 2022
    Configuration menu
    Copy the full SHA
    9c39d8b View commit details
    Browse the repository at this point in the history

Commits on May 6, 2022

  1. Merge remote-tracking branch 'k2-fsa/master'

    @yaozengwei
    yaozengwei committed May 6, 2022
    Configuration menu
    Copy the full SHA
    70634d5 View commit details
    Browse the repository at this point in the history

  2. keep model_avg on cpu

    @yaozengwei
    yaozengwei committed May 6, 2022
    Configuration menu
    Copy the full SHA
    c3bb032 View commit details
    Browse the repository at this point in the history

  3. explicitly convert model_avg to cpu

    @yaozengwei
    yaozengwei committed May 6, 2022
    1 Configuration menu
    Copy the full SHA
    ae50aca View commit details
    Browse the repository at this point in the history

  4. minor fix

    @yaozengwei
    yaozengwei committed May 6, 2022
    Configuration menu
    Copy the full SHA
    3c405fb View commit details
    Browse the repository at this point in the history

  5. remove device convertion for model_avg

    @yaozengwei
    yaozengwei committed May 6, 2022
    Configuration menu
    Copy the full SHA
    a72048b View commit details
    Browse the repository at this point in the history

  6. modify usage of the model device in train.py

    @yaozengwei
    yaozengwei committed May 6, 2022
    Configuration menu
    Copy the full SHA
    dd439b1 View commit details
    Browse the repository at this point in the history

  7. change model.device to next(model.parameters()).device for decoding

    @yaozengwei
    yaozengwei committed May 6, 2022
    Configuration menu
    Copy the full SHA
    b1e9d21 View commit details
    Browse the repository at this point in the history

Commits on May 7, 2022

  1. assert params.start_epoch>0

    @yaozengwei
    yaozengwei committed May 7, 2022
    Configuration menu
    Copy the full SHA
    0bee5d0 View commit details
    Browse the repository at this point in the history

  2. assert params.start_epoch>0, params.start_epoch

    @yaozengwei
    yaozengwei committed May 7, 2022
    Configuration menu
    Copy the full SHA
    5376173 View commit details
    Browse the repository at this point in the history