New Export API

.circleci/config.yml

@@ -197,6 +197,14 @@ jobs:
           command: |
             . venv/bin/activate
             python3 -m pip install --extra-index-url https://pypi.ngc.nvidia.com .[pro]
+
+      - run:
+          name: install pytorch quantization package
+          no_output_timeout: 30m
+          command: |
+            . venv/bin/activate
+            python3 -m pip install pytorch-quantization==2.1.2 --extra-index-url https://pypi.ngc.nvidia.com
+
       - run:
           name: run tests with coverage
           no_output_timeout: 30m

Makefile

@@ -14,3 +14,7 @@ recipe_accuracy_tests:
 	python src/super_gradients/train_from_recipe.py --config-name=cityscapes_regseg48          experiment_name=shortened_cityscapes_regseg48_iou_test   epochs=10 training_hyperparams.average_best_models=False multi_gpu=DDP num_gpus=4
 	python src/super_gradients/examples/convert_recipe_example/convert_recipe_example.py --config-name=cifar10_conversion_params experiment_name=shortened_cifar10_resnet_accuracy_test
 	coverage run --source=super_gradients -m unittest tests/deci_core_recipe_test_suite_runner.py
+
+
+examples_to_docs:
+	jupyter nbconvert --to markdown --output-dir="documentation/source/" --execute src/super_gradients/examples/model_export/models_export.ipynb