작심삼일

[Pytorch 에러] RuntimeError: one_hot is only applicable to index tensor. 본문

정리/잡 에러들

[Pytorch 에러] RuntimeError: one_hot is only applicable to index tensor.

yun_s 2023. 3. 4. 10:29
728x90
반응형

에러

RuntimeError: one_hot is only applicable to index tensor.

F.one_hot 함수를 사용하다 에러가 났다.

해결 방법

one_hot = F.one_hot(vector.to(torch.int64))

dtype을 int로 바꿔주면 된다.

728x90
반응형

'정리 > 잡 에러들' 카테고리의 다른 글

[Python 에러] ValueError: numpy.ndarray size changed, may indicate binary incompatibility. Expected 88 from C header, got 80 from PyObject  (0) 2023.03.08
[Pytorch 에러] RuntimeError: CUDA error: device-side assert triggeredCUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect.For debugging consider passing CUDA_LAUNCH_BLOCKING=1.  (0) 2023.03.04
[Pytorch 에러] AttributeError: cannot assign module before Module.__init__() call  (0) 2023.03.04
[Python 에러] IndentationError: unindent does not match any outer indentation level  (0) 2023.03.04
[opencv 에러] cv2.error: OpenCV(4.5.5) /io/opencv/modules/imgproc/src/color.cpp:182: error: (-215:Assertion failed) !_src.empty() in function 'cvtColor'  (0) 2023.02.24
'정리/잡 에러들' Related Articles more
Comments