mirrors
/
mmocr
mirror of https://github.com/open-mmlab/mmocr.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Enhancement] Remove unnecessary custom_import from train.py (#603)

pull/621/head
Tong Gao 2021-11-25 15:59:35 +08:00 committed by GitHub
parent 809661ea28
commit 1544e9bb29
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tools/train.py
View File

@ -111,10 +111,6 @@ def main():
cfg.data.train.pipeline[0].update( cfg.data.train.pipeline[0].update(
file_client_args=mc['mc_file_client_args']) file_client_args=mc['mc_file_client_args'])
# 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 # set cudnn_benchmark
if cfg.get('cudnn_benchmark', False): if cfg.get('cudnn_benchmark', False):
torch.backends.cudnn.benchmark = True torch.backends.cudnn.benchmark = True