diff --git a/utils/general.py b/utils/general.py index a6639d39480e..aa3d74ed51fb 100755 --- a/utils/general.py +++ b/utils/general.py @@ -38,7 +38,7 @@ # Settings FILE = Path(__file__).resolve() -ROOT = sparseml_root / "src" / "sparseml" / "yolov5" # YOLOv5 root directory +ROOT = sparseml_root / "yolov5" # YOLOv5 root directory if str(ROOT) not in sys.path: sys.path.append(str(ROOT)) # add ROOT to PATH ROOT = Path(os.path.relpath(ROOT, Path.cwd())) # relative