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()

由上述代码知道，是在全连接层后使用，声明变量的位置无所谓，但是使用要在全连接层之后。

Dropout 丢弃函数的使用最先出现在Python成神之路。