Skip to content

Commit 1b7d778

Browse files
authored
[Fix] Fix import error that causes training failure. (#9694)
1 parent ea29a1e commit 1b7d778

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

mmdet/utils/__init__.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
from .memory import AvoidCUDAOOM, AvoidOOM
88
from .misc import find_latest_checkpoint, update_data_root
99
from .replace_cfg_vals import replace_cfg_vals
10+
from .rfnext import rfnext_init_model
1011
from .setup_env import setup_multi_processes
1112
from .split_batch import split_batch
1213
from .util_distribution import build_ddp, build_dp, get_device
@@ -16,5 +17,6 @@
1617
'update_data_root', 'setup_multi_processes', 'get_caller_name',
1718
'log_img_scale', 'compat_cfg', 'split_batch', 'build_ddp', 'build_dp',
1819
'get_device', 'replace_cfg_vals', 'AvoidOOM', 'AvoidCUDAOOM',
19-
'get_max_num_gt_division_factor', 'masked_fill', 'batch_images_to_levels'
20+
'get_max_num_gt_division_factor', 'masked_fill', 'batch_images_to_levels',
21+
'rfnext_init_model'
2022
]

0 commit comments

Comments
 (0)