Results
3
comments of
胡根
请问这个问题解决了吗?
@wangzhedaye 我今天又试了一下,应该是self.w = nn.Parameter(torch.Tensor(config.hidden_size * 2))这行代码有问题, 改成self.w = nn.Parameter(tensor.randn(config.hidden_size * 2)),我跑了多次没发现出现NaN的情况, 还有就是最后的fc层也有一点问题, out = F.relu(out) out = self.fc1(out) out = self.fc(out) # [128, 64] 改成 out = F.relu(self.fc1(out)) out =...
@wangzhedaye 不好意思,打错了,把tensor改成torch就行了,你试一下,看是不是对的