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

Distillation support for torchvision script #1310

Merged
merged 9 commits into from
Jan 11, 2023
Merged

Distillation support for torchvision script #1310

merged 9 commits into from
Jan 11, 2023

Commits on Jan 10, 2023

  1. Add support for self distillation and disable

    @rahul-tuli
    rahul-tuli committed Jan 10, 2023
    Configuration menu
    Copy the full SHA
    c515feb View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into feature/torchvision-distillation-support

    @rahul-tuli
    rahul-tuli committed Jan 10, 2023
    Configuration menu
    Copy the full SHA
    188ae0d View commit details
    Browse the repository at this point in the history

  3. Pull out model creation into a method

    @rahul-tuli
    rahul-tuli committed Jan 10, 2023
    Configuration menu
    Copy the full SHA
    4d8d0e6 View commit details
    Browse the repository at this point in the history

  4. Add support to distill with another model

    @rahul-tuli
    rahul-tuli committed Jan 10, 2023
    Configuration menu
    Copy the full SHA
    33041d8 View commit details
    Browse the repository at this point in the history

  5. Add modifier loss update before backward pass

    @rahul-tuli
    rahul-tuli committed Jan 10, 2023
    Configuration menu
    Copy the full SHA
    11949bd View commit details
    Browse the repository at this point in the history

Commits on Jan 11, 2023

  1. Merge branch 'main' into feature/torchvision-distillation-support

    @rahul-tuli
    rahul-tuli committed Jan 11, 2023
    Configuration menu
    Copy the full SHA
    2b2af45 View commit details
    Browse the repository at this point in the history

  2. bugfix, set loss

    @rahul-tuli
    rahul-tuli committed Jan 11, 2023
    Configuration menu
    Copy the full SHA
    735956e View commit details
    Browse the repository at this point in the history

  3. Merge branch 'main' into feature/torchvision-distillation-support

    @rahul-tuli
    rahul-tuli committed Jan 11, 2023
    Configuration menu
    Copy the full SHA
    6be8dd8 View commit details
    Browse the repository at this point in the history

  4. Update src/sparseml/pytorch/torchvision/train.py 

    Co-authored-by: corey-nm <109536191+corey-nm@users.noreply.github.com>
    @rahul-tuli @corey-nm
    rahul-tuli and corey-nm committed Jan 11, 2023
    Configuration menu
    Copy the full SHA
    2f7cacb View commit details
    Browse the repository at this point in the history