Pytorch中的暹罗神经网络

Question

Pytorch中的暹罗神经网络

pythonpytorch

5

我该如何在PyTorch中实现孪生神经网络？

什么是孪生神经网络？孪生神经网络由两个相同的神经网络组成，每个网络都有一个输入。相同意味着这两个神经网络具有完全相同的结构并共享相同的权重。以下是一个示例图像：

- BiBi

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- BiBi · Accepted Answer

在PyTorch中实现连体神经网络很简单，只需要在不同的输入上两次调用网络函数即可。

mynet = torch.nn.Sequential(
        nn.Linear(10, 512),
        nn.ReLU(),
        nn.Linear(512, 2))
...
output1 = mynet(input1)
output2 = mynet(input2)
...
loss.backward()

当调用loss.backward()时，PyTorch会自动汇总来自两个mynet的梯度。您可以在这里找到一个完整的示例。