Pytorch 中的孿生神經網絡

Question

如何在 PyTorch 中實現孿生神經網絡？

什么是孿生神經網絡？ 孿生神經網絡由兩個相同的神經網絡組成，每個神經網絡接受一個輸入。 相同意味着兩個神經網絡具有完全相同的架構並共享相同的權重。

Answer 1

在 PyTorch 中實現孿生神經網絡就像在不同輸入上調用網絡函數兩次一樣簡單。

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

調用loss.backwad() ，PyTorch 會自動將來自兩次調用mynet的梯度mynet 。

您可以在此處找到完整的示例。