def forward(self, x, reverse=False):
if reverse:
x = grad_reverse(x, self.lambd)
x = F.relu(self.bn2_fc(self.fc2(x)))
x = F.dropout(x, p = 0.5)
x = self.fc3(x)
return x
def forward(self, x, reverse=False):
if reverse:
x = grad_reverse(x, self.lambd)
x = x.view(x.size(0), 6400)
x = self.relu(self.bn1_fc(self.fc1(x)))
x = F.dropout(x, training=self.training)
x = self.fc2(x)
return x
def forward(self, x, reverse=False):
if reverse:
x = grad_reverse(x, self.lambd)
x = F.dropout(x, training=self.training, p=self.prob)
x = F.relu(self.bn1_fc(self.fc1(x)))
x = F.dropout(x, training=self.training, p=self.prob)
x = F.relu(self.bn2_fc(self.fc2(x)))
x = F.dropout(x, training=self.training, p=self.prob)
x = self.fc3(x)
return x
def forward(self, x, mode, reverse=False):
if mode != 'ad_drop':
if reverse:
rev = grad_reverse.grad_reverse()
x = rev(x)
else:
x = F.relu(self.bn1_fc(self.fc1(x)))
x = F.dropout(x, training=self.training, p=self.prob)
x = F.relu(self.bn2_fc(self.fc2(x)))
if mode == 'ad_drop':
x = F.dropout(x, training=self.training, p=self.prob)
x = self.fc3(x)
return x