def parse_batch_train(self, batch_x, batch_u):
    input_x = batch_x['img']
    label_x = batch_x['label']
    input_u = batch_u['img']
    input_x = input_x.to(self.device)
    label_x = label_x.to(self.device)
    input_u = input_u.to(self.device)
    return (input_x, label_x, input_u)
def parse_batch_train(self, batch_x, batch_u):
    input_x = batch_x['img']
    label_x = batch_x['label']
    input_u = batch_u['img']
    input_x = input_x.to(self.device)
    label_x = label_x.to(self.device)
    input_u = input_u.to(self.device)
    return (input_x, label_x, input_u)