Skip to content

Commit 9251dc1

Browse files
glenn-jochertdhooghe
authored andcommitted
Add check_yaml() to benchmarks.py (ultralytics#7916)
Locate file in subdirectories.
1 parent f4be40d commit 9251dc1

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

utils/benchmarks.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
import export
4242
import val
4343
from utils import notebook_init
44-
from utils.general import LOGGER, print_args
44+
from utils.general import LOGGER, check_yaml, print_args
4545
from utils.torch_utils import select_device
4646

4747

@@ -136,6 +136,7 @@ def parse_opt():
136136
parser.add_argument('--test', action='store_true', help='test exports only')
137137
parser.add_argument('--pt-only', action='store_true', help='test PyTorch only')
138138
opt = parser.parse_args()
139+
opt.data = check_yaml(opt.data) # check YAML
139140
print_args(vars(opt))
140141
return opt
141142

0 commit comments

Comments
 (0)