sparseml icon indicating copy to clipboard operation
sparseml copied to clipboard

Published 20 hours ago • verified publisher icon neuralmagic

update get_named_layers_and_params_by_regex in src/sparseml/pytorch/u…

Open ohaijen opened this issue 2 years ago • 0 comments

Update get_named_layers_and_params_by_regex in src/sparseml/pytorch/utils/helpers.py to still match in the DDP training case, where module. is prepended to layer names. I came across this issue when trying to run the suggested training command in src/sparseml/pytorch/torchvision on multiple GPUs.

ohaijen avatar May 18 '23 18:05 ohaijen