728x90
labels = labels.unsqueeze(1)
추가하여 해결
728x90
'ERROR' 카테고리의 다른 글
[ML] ValueError: only one element tensors can be converted to Python scalars (0) | 2023.01.10 |
---|---|
[ML] RuntimeError: Found dtype Long but expected Float (0) | 2023.01.09 |
[ML] 'int' object has no attribute 'to' (0) | 2023.01.09 |
[GAN] Loss Function이 극값이 나옴, Oscillation (0) | 2023.01.05 |
[Plot] TypeError: Image data of dtype object cannot be converted to float (1) | 2022.12.30 |