sparseml icon indicating copy to clipboard operation
sparseml copied to clipboard
verified publisher icon neuralmagic

Refactor DeepSparse related functions out of Onnx

Open kevinaer opened this issue 4 years ago • 1 comments

Changes:

  • Moved DeepSparseModelRunner and DeepSparseModelRunner to sparseml.deepsparse.utils.models
  • Moved pruning_loss_sens_one_shot and pruning_perf_sens_one_shot to sparseml.deepsparse.optim.sensitivity_pruning
  • Moved tests related to above to appropriate packages

kevinaer avatar Sep 07 '21 16:09 kevinaer

@kevinaer code looks good, can you go through and validate that tests are running through for you locally?

markurtz avatar Sep 23 '21 14:09 markurtz