search

JiaquanYe / TableMASTER-mmocr

2nd solution of ICDAR 2021 Competition on Scientific Literature Parsing, Task B.
Apache License 2.0
410 stars 100 forks source link

怀疑这里有问题 #37

Open little-polka-dot opened 2 years ago

little-polka-dot commented 2 years ago

master_decoder.py 271~272行

for layer in self.cls_layer: cls_x = layer(x, feature, src_mask, tgt_mask)

改成

cls_x = x for layer in self.cls_layer: cls_x = layer(cls_x, feature, src_mask, tgt_mask)

其他地方也有类似问题

JiaquanYe commented 2 years ago

Yes, you're right. It's doesn't go wrong because of 'self.cls_layer' length is 1. Your solution is more reasonable.