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 ModelPruning(make_pruning_permanent=True) buffers getting removed when saved during training #6073

Merged
merged 11 commits into from
Mar 3, 2021
Merged

Fix ModelPruning(make_pruning_permanent=True) buffers getting removed when saved during training #6073

merged 11 commits into from
Mar 3, 2021

Commits on Feb 19, 2021

  1. Add checkpoint parameter to on_save_checkpoint

    @carmocca
    carmocca committed Feb 19, 2021
    Configuration menu
    Copy the full SHA
    ab44327 View commit details
    Browse the repository at this point in the history

  2. Update CHANGELOG

    @carmocca
    carmocca committed Feb 19, 2021
    Configuration menu
    Copy the full SHA
    8fa669d View commit details
    Browse the repository at this point in the history

  3. Do not prune permanently during training but save the pruned model

    @carmocca
    carmocca committed Feb 19, 2021
    Configuration menu
    Copy the full SHA
    449f427 View commit details
    Browse the repository at this point in the history

  4. Update CHANGELOG

    @carmocca
    carmocca committed Feb 19, 2021
    Configuration menu
    Copy the full SHA
    7763324 View commit details
    Browse the repository at this point in the history

Commits on Feb 21, 2021

  1. Move to cpu before copy

    @carmocca
    carmocca committed Feb 21, 2021
    Configuration menu
    Copy the full SHA
    9f04f74 View commit details
    Browse the repository at this point in the history

Commits on Feb 22, 2021

  1. Apply suggestions from code review 

    Co-authored-by: chaton <thomas@grid.ai>
    @carmocca @tchaton
    carmocca and tchaton committed Feb 22, 2021
    Configuration menu
    Copy the full SHA
    706e848 View commit details
    Browse the repository at this point in the history

Commits on Feb 25, 2021

  1. Merge branch 'master' into pruning-modelcheckpoint-hotfix

    @carmocca
    carmocca committed Feb 25, 2021
    Configuration menu
    Copy the full SHA
    90f77f3 View commit details
    Browse the repository at this point in the history

Commits on Mar 1, 2021

  1. Address comments

    @carmocca
    carmocca committed Mar 1, 2021
    Configuration menu
    Copy the full SHA
    ff6fa73 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'master' into pruning-modelcheckpoint-hotfix

    @carmocca
    carmocca committed Mar 1, 2021
    Configuration menu
    Copy the full SHA
    e04b1a5 View commit details
    Browse the repository at this point in the history

  3. Move to correct section

    @carmocca
    carmocca committed Mar 1, 2021
    Configuration menu
    Copy the full SHA
    ed364a7 View commit details
    Browse the repository at this point in the history

  4. Merge branch 'master' into pruning-modelcheckpoint-hotfix

    @carmocca
    carmocca committed Mar 1, 2021
    Configuration menu
    Copy the full SHA
    e663641 View commit details
    Browse the repository at this point in the history