파이토치 오류

[111 (자ㅐ우ㅐ)]

안녕하세요 파이토치 loss 로 ssim 으로 사용하던 도중 오류가 났습니다. 혹시 해결할수있을까요
추가로 입력을 tensor 로 사용하다 중간에 np 형태로 바꿔줄수있나요?

[9bow (박정환)]

안녕하세요,

가장 아래쪽의 TypeError 부분의 에러메시지를 읽어보시면,
(GPU 상의) CUDA tensor를 바로 ndarray(NumPy array)로 변환하지 못해서 발생하는 에러 메시지임을 확인하실 수 있습니다.

따라서 ssim()의 입력들 중 (GPU 상의) CUDA tensor를 .cpu()로 (CPU 상의) tensor로 변환해주시면 될 것 같습니다.

아래 문서들을 참고해보셔도 좋을 것 같습니다:

• torch.Tensor — PyTorch 1.9.0 documentation
• torch.Tensor.cpu — PyTorch 1.9.0 documentation
• Can't convert CUDA tensor to numpy. Use Tensor.cpu() to copy the tensor to host memory first - #5 by lavender99 - PyTorch Forums