ustcJin
ustcJin
这个DSSM没有 embedding啊，直接就接了一个全连接层，有问题吧？
> 我们只需要一个数值进行排序，这里计算的是yes的概率，进行排序。 ok 明白，那最后算loss的时候，target为啥都是0，batch内第一条是pos，其余是neg， target应该是[1, 0, 0, 0...]才对啊 grouped_logits = ranker_logits.view(self.train_batch_size, -1) target = torch.zeros(self.train_batch_size, device=grouped_logits.device, dtype=torch.long) loss = self.compute_loss(grouped_logits, target)
> target指示的是第几个为pos，所以0的意思是第0个为pos 嗯嗯，我搞混了，这个batch 不是 group的那个，应该是 多个group组成batch，每个group内Yes最大概率是在第0个