Unverified 提交 15057d2a authored 作者: Glenn Jocher's avatar Glenn Jocher 提交者: GitHub

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

Locate file in subdirectories.
上级 5774a151
...@@ -41,7 +41,7 @@ if str(ROOT) not in sys.path: ...@@ -41,7 +41,7 @@ if str(ROOT) not in sys.path:
import export import export
import val import val
from utils import notebook_init 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 from utils.torch_utils import select_device
...@@ -136,6 +136,7 @@ def parse_opt(): ...@@ -136,6 +136,7 @@ def parse_opt():
parser.add_argument('--test', action='store_true', help='test exports only') parser.add_argument('--test', action='store_true', help='test exports only')
parser.add_argument('--pt-only', action='store_true', help='test PyTorch only') parser.add_argument('--pt-only', action='store_true', help='test PyTorch only')
opt = parser.parse_args() opt = parser.parse_args()
opt.data = check_yaml(opt.data) # check YAML
print_args(vars(opt)) print_args(vars(opt))
return opt return opt
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论