Nn.dropout

Batchnorm1d pytorch

class network(nn.Module):
    def __init__(self):
        super(network, self).__init__()
        self.linear1 = nn.Linear(in_features=40, out_features=320)
        self.bn1 = nn.BatchNorm1d(num_features=320)
        self.linear2 = nn.Linear(in_features=320, out_features=2)

    def forward(self, input):  # Input is a 1D tensor
        y = F.relu(self.bn1(self.linear1(input)))
        y = F.softmax(self.linear2(y), dim=1)
        return y
    
model = network()
x = torch.randn(10, 40)
output = model(x)

Source: discuss.pytorch.org

Nn.dropout

>>> m = nn.Dropout(p=0.2)
>>> input = torch.randn(20, 16)
>>> output = m(input)

Source: pytorch.org

Similar codes

-1 vote181 views1 answer
Why women are better than men
Python
0 vote193 views1 answer
Error 302 heroku django
Python
22 votes134 views2 answers
Range() function
Python
12 votes182 views2 answers
Matplotlib legend out of plot
Python
61 votes187 views6 answers
Remove from string python
Python
0 vote200 views1 answer
Boto3 with aws profile
Python
0 vote178 views2 answers
Django run command from code
Python
0 vote180 views1 answer
Python insert sorted
Python
0 vote163 views1 answer
Python program for getting url, hostname, port numbers
Python
8 votes183 views2 answers
Pandas filter rows by value in list
Python