glenn-jocher commited on
Commit
15057d2
·
unverified ·
1 Parent(s): 5774a15

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

Browse files
Files changed (1) hide show
  1. 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