Adds back a required argument in default.yaml (#1620)

Updates the hotfix version
neuralmagic · Jun 9, 2023 · 9f7606c · 9f7606c
1 parent 8422758
commit 9f7606c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/src/sparseml/version.py b/src/sparseml/version.py
@@ -19,7 +19,7 @@
 from datetime import date
 
 
-version_base = "1.5.0"
+version_base = "1.5.1"
 is_release = False  # change to True to set the generated version as a release version
 
 

diff --git a/src/sparseml/yolov8/default.yaml b/src/sparseml/yolov8/default.yaml
@@ -2,6 +2,7 @@
 
 recipe: null
 recipe_args: null
+datasets_dir: null
 
 task: "detect" # choices=['detect', 'segment', 'classify', 'init'] # init is a special case. Specify task to run.
 mode: "train" # choices=['train', 'val', 'predict'] # mode to run task in.