Add check_yaml() to benchmarks.py (#7916)
Locate file in subdirectories.
This commit is contained in:
1 parent
5774a1514d
commit
15057d2a34
1 file changed
+2
-1
+2
-1
@@ -41,7 +41,7 @@ if str(ROOT) not in sys.path:
|
||||
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
|
||||
|
||||
|
||||
Reference in new issue
Block a user