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

Enable running pytorch.torchvision.train with distributed data parallel #1698

Merged
merged 9 commits into from
Aug 7, 2023
Merged

Enable running pytorch.torchvision.train with distributed data parallel #1698

merged 9 commits into from
Aug 7, 2023

Commits on May 18, 2023

  1. update get_named_layers_and_params_by_regex in src/sparseml/pytorch/u… 

    …tils/helpers.py to still match in the DDP training case, where module. is prepended to layer names
    @ohaijen
    ohaijen committed May 18, 2023
    Configuration menu
    Copy the full SHA
    d7dc46a View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' of github.com:neuralmagic/sparseml into main

    @ohaijen
    ohaijen committed May 18, 2023
    Configuration menu
    Copy the full SHA
    86703b9 View commit details
    Browse the repository at this point in the history

Commits on Jul 17, 2023

  1. Merge branch 'main' of github.com:neuralmagic/sparseml into main

    @ohaijen
    ohaijen committed Jul 17, 2023
    Configuration menu
    Copy the full SHA
    51d7453 View commit details
    Browse the repository at this point in the history

Commits on Aug 2, 2023

  1. Merge branch 'main' of github.com:neuralmagic/sparseml into main

    @ohaijen
    ohaijen committed Aug 2, 2023
    Configuration menu
    Copy the full SHA
    ffc3eac View commit details
    Browse the repository at this point in the history

  2. make pytorch.torchvision.train.py work with torch.distributed.launch

    @ohaijen
    ohaijen committed Aug 2, 2023
    Configuration menu
    Copy the full SHA
    516a8f6 View commit details
    Browse the repository at this point in the history

  3. undo accidental add

    @ohaijen
    ohaijen committed Aug 2, 2023
    Configuration menu
    Copy the full SHA
    c0051af View commit details
    Browse the repository at this point in the history

  4. expand comment and fix for CLI

    @ohaijen
    ohaijen committed Aug 2, 2023
    Configuration menu
    Copy the full SHA
    fc22730 View commit details
    Browse the repository at this point in the history

  5. make quality

    @ohaijen
    ohaijen committed Aug 2, 2023
    Configuration menu
    Copy the full SHA
    878e79a View commit details
    Browse the repository at this point in the history

Commits on Aug 3, 2023

  1. fix overly long line

    @ohaijen
    ohaijen committed Aug 3, 2023
    Configuration menu
    Copy the full SHA
    5e3ce2b View commit details
    Browse the repository at this point in the history