Add `check_yaml()` to benchmarks.py (#7916)
Browse files- utils/benchmarks.py +2 -1
utils/benchmarks.py
CHANGED
@@ -41,7 +41,7 @@ if str(ROOT) not in sys.path:
|
|
41 |
import export
|
42 |
import val
|
43 |
from utils import notebook_init
|
44 |
-
from utils.general import LOGGER, print_args
|
45 |
from utils.torch_utils import select_device
|
46 |
|
47 |
|
@@ -136,6 +136,7 @@ def parse_opt():
|
|
136 |
parser.add_argument('--test', action='store_true', help='test exports only')
|
137 |
parser.add_argument('--pt-only', action='store_true', help='test PyTorch only')
|
138 |
opt = parser.parse_args()
|
|
|
139 |
print_args(vars(opt))
|
140 |
return opt
|
141 |
|
|
|
41 |
import export
|
42 |
import val
|
43 |
from utils import notebook_init
|
44 |
+
from utils.general import LOGGER, check_yaml, print_args
|
45 |
from utils.torch_utils import select_device
|
46 |
|
47 |
|
|
|
136 |
parser.add_argument('--test', action='store_true', help='test exports only')
|
137 |
parser.add_argument('--pt-only', action='store_true', help='test PyTorch only')
|
138 |
opt = parser.parse_args()
|
139 |
+
opt.data = check_yaml(opt.data) # check YAML
|
140 |
print_args(vars(opt))
|
141 |
return opt
|
142 |
|