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