[ML] RuntimeError: Found dtype Long but expected Float

labels = labels.to(config.device).to(torch.float32)

.to(torch.float32) 추가하여 해결