[Enhancement] Remove unnecessary custom_import from test.py (#758)

pull/755/head
liukuikun 2022-01-27 11:02:16 +08:00 committed by GitHub
parent e0ff11819b
commit b054f609b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 4 deletions

View File

@ -130,10 +130,7 @@ def main():
if args.cfg_options is not None:
cfg.merge_from_dict(args.cfg_options)
setup_multi_processes(cfg)
# import modules from string list.
if cfg.get('custom_imports', None):
from mmcv.utils import import_modules_from_strings
import_modules_from_strings(**cfg['custom_imports'])
# set cudnn_benchmark
if cfg.get('cudnn_benchmark', False):
torch.backends.cudnn.benchmark = True