def forward(self, input_data, alpha):
input_data = input_data.expand(input_data.data.shape[0], 3, 28, 28)
feature = self.feature(input_data)
feature = feature.view(-1, 50 * 4 * 4)
reverse_feature = ReverseLayerGrad.apply(feature, alpha)
class_output = self.class_classifier(feature)
domain_output = self.domain_classifier(reverse_feature)
return class_output, domain_output
def forward(self, input, alpha):
input = self.features(input)
features = input.view(input.size(0), -1)
input = self.classifier(features)
reverse_features = ReverseLayerGrad.apply(features, alpha)
class_output = self.cls_fc(input)
domain_output = self.discriminator(reverse_features)
return class_output, domain_output
def forward(self, input, delta):
input = self.features(input)
features = input.view(input.size(0), -1)
input = self.classifier(features)
reverse_features = ReverseLayerGrad.apply(features, delta)
class_pred = self.cls_fc(input)
loss_adv = self.discriminator(reverse_features)
return class_pred, loss_adv