only in DRRG import DRRGHead
@@ -24,7 +24,6 @@ def build_head(config):
from .det_fce_head import FCEHead
from .e2e_pg_head import PGHead
from .det_ct_head import CT_Head
- from .det_drrg_head import DRRGHead
# rec head
from .rec_ctc_head import CTCHead
@@ -59,6 +58,10 @@ def build_head(config):
'DRRGHead'
]
+ if config['name'] == 'DRRGHead':
+ from .det_drrg_head import DRRGHead
+ support_dict.append('DRRGHead')
+
#table head
module_name = config.pop('name')
