Add check_yaml() to benchmarks.py (#7916)

Locate file in subdirectories.
ultralytics · May 21, 2022 · 15057d2 · 15057d2
1 parent 5774a15
commit 15057d2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/utils/benchmarks.py b/utils/benchmarks.py
@@ -41,7 +41,7 @@
 import export
 import val
 from utils import notebook_init
-from utils.general import LOGGER, print_args
+from utils.general import LOGGER, check_yaml, print_args
 from utils.torch_utils import select_device
 
 
@@ -136,6 +136,7 @@ def parse_opt():
     parser.add_argument('--test', action='store_true', help='test exports only')
     parser.add_argument('--pt-only', action='store_true', help='test PyTorch only')
     opt = parser.parse_args()
+    opt.data = check_yaml(opt.data)  # check YAML
     print_args(vars(opt))
     return opt