We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
check_yaml()
1 parent f4be40d commit 9251dc1Copy full SHA for 9251dc1
utils/benchmarks.py
@@ -41,7 +41,7 @@
41
import export
42
import val
43
from utils import notebook_init
44
-from utils.general import LOGGER, print_args
+from utils.general import LOGGER, check_yaml, 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
+ opt.data = check_yaml(opt.data) # check YAML
140
print_args(vars(opt))
141
return opt
142
0 commit comments