tbwxmu/SAMPN

提问:网络框架图与代码的对应关系

Opened this issue · 0 comments

1.网络框架图中 self attention layer 是否 对应于代码mpn.py 中的self.attention = args.attention语句?
2.fully connect layer 是否对应于代码
if self.attention:
self.W_a = nn.Linear(self.hidden_size, self.hidden_size, bias=self.bias)
self.W_b = nn.Linear(self.hidden_size, self.hidden_size)