add comment

pull/2333/head
MengzhangLI 2022-11-22 22:38:10 +08:00
parent 589d8ac224
commit 91440718de
1 changed files with 3 additions and 0 deletions

View File

@ -411,6 +411,9 @@ class IterativeDecodeHead(BaseDecodeHead):
def __init__(self, num_stages, kernel_generate_head, kernel_update_head,
**kwargs):
# ``IterativeDecodeHead`` would skip initialization of
# ``BaseDecodeHead`` which would be called when building
# ``self.kernel_generate_head``.
super(BaseDecodeHead, self).__init__(**kwargs)
assert num_stages == len(kernel_update_head)
self.num_stages = num_stages