Browse Source

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

Locate file in subdirectories.
modifyDataloader
Glenn Jocher GitHub 2 years ago
parent
commit
15057d2a34
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      utils/benchmarks.py

+ 2
- 1
utils/benchmarks.py 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


Loading…
Cancel
Save