model = nn.DataParallel(Neural())
model = model.to(torch.device('cuda'))
images_first, images_second = (images_first.to(torch.device('cuda'), torch.float32), images_second.to(torch.device('cuda'), torch.float32))
        
output = model(images_first, images_second) <--- here I have problem