반응형
가끔 가다가
RuntimeError: expected scalar type Float but found Double
이런 에러가 발생한다.
그럴경우...
a 라는 텐서가 있으면 아래와 같이 케스팅해주면 된다.
type도 변환된 자기 자신을 리턴한다.
a = a.type(torch.FloatTensor).to(device)
반응형
'Data-science > deep learning' 카테고리의 다른 글
| [nvidia-docker] pytorch 실행시 메모리 문제 (0) | 2021.03.08 |
|---|---|
| [nvidia-docker] NGC 설치로 pytorch build 필요없이 실행하기 - nvidia imaginaire 이용하는 법 (14) | 2021.03.08 |
| nvidia-smi랑 nvcc --version이 다를 때 (0) | 2021.01.24 |
| Closed-Form Factorization of Latent Semantics in GANs 논문 설명 (0) | 2021.01.23 |
| [pytorch] inference시 memory leak, 메모리 과부하 문제 (6) | 2021.01.11 |
