add comment

This commit is contained in:
MengzhangLI 2022-11-22 22:42:11 +08:00
parent 58e73c3fab
commit 8b61820751
2 changed files with 4 additions and 0 deletions

1
data Symbolic link
View File

@ -0,0 +1 @@
/home/PJLAB/limengzhang/Downloads/datasets/

View File

@ -413,6 +413,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