Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
This PR adds environment checks that will help detect a broken sparsify environment before a run is fully launched. There are three pathways by which environment checks are launched:
sparsify.check_environment
will run all health checkssparsify.run training-aware
orsparsify.run sparse-transfer
will run health checks relevant to these two pathwayssparsify.run one-shot
will run the health checks relevant to one-shotCurrently implemented health checks
GPU device check - this is a non-failing check, but it will raise a loud error
ORT health check - this is a failing check which will point the user to the ORT-GPU dependency matrix
Test Plan
sparsify.check_environment
CUDA_VISIBLE_DEVICES="" sparsify.check_environment
sparsify.run training-aware --use-case image-classification --model mobilenet-v1 --data /network/datasets/imagenet --optim-level 0
sparsify.run one-shot --model model.onnx --data coco_calibration --use-case cv-detection --optim-level 0.5