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

2021-11-25 15:59:35 +08:00 · 2021-11-25 15:59:35 +08:00 · 1544e9bb29
parent 809661ea28
commit 1544e9bb29
1 changed files with 0 additions and 4 deletions
--- a/tools/train.py
+++ b/tools/train.py
@ -111,10 +111,6 @@ def main():
            cfg.data.train.pipeline[0].update(
                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
    if cfg.get('cudnn_benchmark', False):
        torch.backends.cudnn.benchmark = True