text: import torch
  import torch.nn as nn
  from loss_funcs import *
  
  class TransferLoss(nn.Module):
      def __init__(self, loss_type, **kwargs):
          super(TransferLoss, self).__init__()
          if loss_type == "mmd":
              self.loss_func = MMDLoss(**kwargs)
          elif loss_type == "lmmd":
              self.loss_func = LMMDLoss(**kwargs)
          elif loss_type == "coral":
              self.loss_func = CORAL
          elif loss_type == "adv":
              self.loss_func = AdversarialLoss(**kwargs)
          else:
              print("WARNING: No valid transfer loss function is used.")
              self.loss_func = lambda x, y: 0 # return 0
      
      def forward(self, source, target, **kwargs):
          return self.loss_func(source, target, **kwargs)