繁体   English   中英

如何使用 pytorch 向神经网络添加偏差?

[英]How can I add bias using pytorch to a neural network?

如果我有这个神经网络(NN):

 class NeuralNet(nn.Module):
    def __init__(self, input_size, hidden_size, num_classes, p=dropout):
        super(NeuralNet, self).__init__()
        self.fc1 = nn.Linear(input_size, hidden_size)
        self.fc2 = nn.Linear(hidden_size, hidden_size)
        self.fc3 = nn.Linear(hidden_size, num_classes)
        self.dropout = nn.Dropout(p=p)

    def forward(self, x):
        out = F.relu(self.fc1(x))
        out = F.relu(  self.dropout(self.fc2(out)))
        out = self.fc3(out)
        return out

你如何给它添加偏见? 例如,在隐藏层上。 我在网上找不到任何东西。

这取决于您使用的图层。 有些人没有这个选项。 例如,在线性中,您可以使用:

self.fc1 = nn.Linear(input_size, hidden_size, bias =False) # Either true or false, the default is true.

文档中,您可以看到其他类型的图层是否有偏差选项。

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM