diff --git a/utils/benchmarks.py b/utils/benchmarks.py index c3636b9e4df4..99910050d0f7 100644 --- 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