def parse_batch_train(self, batch):
    input = batch['img']
    label = batch['label']
    domain = batch['domain']
    input = input.to(self.device)
    label = label.to(self.device)
    domain = domain.to(self.device)
    return (input, label, domain)
def parse_batch_train(self, batch):
    input = batch['img']
    label = batch['label']
    domain = batch['domain']
    input = input.to(self.device)
    label = label.to(self.device)
    domain = domain.to(self.device)
    return (input, label, domain)