Fixes for 3.4.1 release (#1629)

* updated export recipe

* updated fixes for release

* Update versions in notebooks

* Update versions in notebooks

---------

Co-authored-by: Eugene <ekhvedchenya@gmail.com>

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"
    ]
   },
   {

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"
    ]
   },
   {

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"
    ]
   },
   {

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",

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

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

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

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

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,

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()