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

Locate file in subdirectories.
pull/7921/head
Glenn Jocher 2022-05-21 19:39:19 +02:00 committed by GitHub
parent 5774a1514d
commit 15057d2a34
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -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