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 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():
|
|||||||
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
|
||||||
|
|
||||||
|
|||||||
Reference in new issue
Block a user