pytorch nn.Dropout使用

Class USeDropout(nn.Module):

    def __init__(self):

        super(DropoutFC, self).__init__()

        self.fc = nn.Linear(100,20)

        self.dropout = nn.Dropout(p=0.5)

 
    def forward(self, input):

        out = self.fc(input)

        out = self.dropout(out)

        return out

Net = USeDropout()

Net.train()

 

 示例代码如上,直接调用nn.Dropout即可,但是注意在调用时要将模型参数传入。

 

你可能感兴趣的:(pytorch使用,pytorch,神经网络,深度学习,机器学习,数据挖掘)