프로그래밍/기타

[오류해결] RuntimeError: one_hot is only applicable to index tensor.

흔한티벳여우 2022. 6. 21. 18:43
반응형

pytorch 를 사용하다 one_hot 인코딩을 사용 중에 오류가 발생하였다.

 

원인은 dtype이 float64였기 때문이였다.

int64로 바꾸어주면 문제가 해결된다.

 

t = F.one_hot(t.to(torch.int64), num_classes = 201)

 

반응형