提交 8123688a 编写于 作者: A andyjpaddle

update rec_sar_head

上级 f57a6d1b
...@@ -151,7 +151,7 @@ class ParallelSARDecoder(BaseDecoder): ...@@ -151,7 +151,7 @@ class ParallelSARDecoder(BaseDecoder):
**kwargs): **kwargs):
super().__init__() super().__init__()
self.num_classes = num_classes self.num_classes = out_channels
self.enc_bi_rnn = enc_bi_rnn self.enc_bi_rnn = enc_bi_rnn
self.d_k = d_k self.d_k = d_k
self.start_idx = out_channels - 2 self.start_idx = out_channels - 2
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册