728x90
labels = labels.to(config.device).to(torch.float32)
.to(torch.float32) 추가하여 해결
728x90
'ERROR' 카테고리의 다른 글
728x90
labels = labels.to(config.device).to(torch.float32)
.to(torch.float32) 추가하여 해결
728x90