diff --git a/notebooks/PTQ_and_QAT_for_classification.ipynb b/notebooks/PTQ_and_QAT_for_classification.ipynb index e96fe54623..fed0a95224 100644 --- a/notebooks/PTQ_and_QAT_for_classification.ipynb +++ b/notebooks/PTQ_and_QAT_for_classification.ipynb @@ -61,7 +61,7 @@ }, "outputs": [], "source": [ - "!pip -qq install super-gradients==3.4.0 pytorch-quantization==2.1.2 --extra-index-url https://pypi.ngc.nvidia.com" + "!pip -qq install super-gradients==3.4.1 pytorch-quantization==2.1.2 --extra-index-url https://pypi.ngc.nvidia.com" ] }, { diff --git a/notebooks/how_to_use_knowledge_distillation_for_classification.ipynb b/notebooks/how_to_use_knowledge_distillation_for_classification.ipynb index 4c05ed5f7f..1f2a9ccdd4 100644 --- a/notebooks/how_to_use_knowledge_distillation_for_classification.ipynb +++ b/notebooks/how_to_use_knowledge_distillation_for_classification.ipynb @@ -34,7 +34,7 @@ }, "outputs": [], "source": [ - "!pip install -qq super_gradients==3.4.0" + "!pip install -qq super_gradients==3.4.1" ] }, { diff --git a/notebooks/transfer_learning_classification.ipynb b/notebooks/transfer_learning_classification.ipynb index b8be6e6f2f..7ffaf02f6c 100644 --- a/notebooks/transfer_learning_classification.ipynb +++ b/notebooks/transfer_learning_classification.ipynb @@ -59,7 +59,7 @@ } ], "source": [ - "! pip install -qq super-gradients==3.4.0" + "! pip install -qq super-gradients==3.4.1" ] }, { diff --git a/notebooks/what_are_recipes_and_how_to_use.ipynb b/notebooks/what_are_recipes_and_how_to_use.ipynb index 920c31a4f0..73ff2c77f1 100644 --- a/notebooks/what_are_recipes_and_how_to_use.ipynb +++ b/notebooks/what_are_recipes_and_how_to_use.ipynb @@ -41,7 +41,7 @@ { "cell_type": "code", "source": [ - "!pip install -q super-gradients==3.4.0" + "!pip install -q super-gradients==3.4.1" ], "metadata": { "id": "8uZM-4va5Rpu", diff --git a/notebooks/yolo_nas_pose_eval_with_pycocotools.ipynb b/notebooks/yolo_nas_pose_eval_with_pycocotools.ipynb index 36e3e5078e..4ff8dbd9cb 100644 --- a/notebooks/yolo_nas_pose_eval_with_pycocotools.ipynb +++ b/notebooks/yolo_nas_pose_eval_with_pycocotools.ipynb @@ -36,7 +36,7 @@ "execution_count": null, "outputs": [], "source": [ - "!pip install -qq super_gradients==3.4.0" + "!pip install -qq super_gradients==3.4.1" ], "metadata": { "collapsed": false diff --git a/requirements.txt b/requirements.txt index d5f4b63f35..10ddcde898 100755 --- a/requirements.txt +++ b/requirements.txt @@ -36,4 +36,4 @@ numpy<=1.23 rapidfuzz json-tricks==3.16.1 onnx-simplifier>=0.3.6,<1.0 -data-gradients>=0.2.2 +data-gradients>=0.3.0 diff --git a/src/super_gradients/__init__.py b/src/super_gradients/__init__.py index 7d05d73a93..06522510a3 100755 --- a/src/super_gradients/__init__.py +++ b/src/super_gradients/__init__.py @@ -1,4 +1,4 @@ -__version__ = "3.4.0" +__version__ = "3.4.1" from super_gradients.common import init_trainer, is_distributed, object_names from super_gradients.training import losses, utils, datasets_utils, DataAugmentation, Trainer, KDTrainer, QATTrainer diff --git a/src/super_gradients/examples/model_export/models_export.ipynb b/src/super_gradients/examples/model_export/models_export.ipynb index cbacc65621..d8c4d92e4b 100644 --- a/src/super_gradients/examples/model_export/models_export.ipynb +++ b/src/super_gradients/examples/model_export/models_export.ipynb @@ -49,7 +49,7 @@ "execution_count": null, "outputs": [], "source": [ - "!pip install -qq super_gradients==3.4.0" + "!pip install -qq super_gradients==3.4.1" ], "metadata": { "collapsed": false diff --git a/src/super_gradients/examples/model_export/models_export_pose.ipynb b/src/super_gradients/examples/model_export/models_export_pose.ipynb index 2cb31fed2f..28f251af43 100644 --- a/src/super_gradients/examples/model_export/models_export_pose.ipynb +++ b/src/super_gradients/examples/model_export/models_export_pose.ipynb @@ -58,7 +58,7 @@ "execution_count": 24, "outputs": [], "source": [ - "!pip install -qq super-gradients==3.4.0" + "!pip install -qq super-gradients==3.4.1" ], "metadata": { "collapsed": false, diff --git a/src/super_gradients/scripts/export_recipe.py b/src/super_gradients/scripts/export_recipe.py index d4ae3aaaf9..34ca35a213 100644 --- a/src/super_gradients/scripts/export_recipe.py +++ b/src/super_gradients/scripts/export_recipe.py @@ -25,11 +25,13 @@ import argparse import os +import pkg_resources + from super_gradients.common.environment.cfg_utils import export_recipe if __name__ == "__main__": parser = argparse.ArgumentParser() - parser.add_argument("--config_dir", type=str, default=None, help="The config directory path") + parser.add_argument("--config_dir", type=str, default=pkg_resources.resource_filename("super_gradients.recipes", ""), help="The config directory path") parser.add_argument("config_name", type=str, help=".yaml filename") parser.add_argument("save_path", type=str, default=None, help="Destination path to the output .yaml file") args = parser.parse_args()