[Tensor 오류] RuntimeError : expected scalar type Double but found Float

RuntimeError : expected scalar type Double but found Float

 

 

파이토치를 쓰다보면 종종 자주 등장하는 경고문구인데..

Pytorch는 Tensor연산시 Float32를 기준으로 진행하기에

본인이 학습시키고자 하는 데이터의 형식이 double이거나, Float64, int같은 다른 형태일때 발생하는 오류입니다..

 

DataFrame이나 numpy로 그냥 불러오면 double인 경우도 있는데요

이럴때는 하기와 같이 data type을 Float32로 바꿔주면 정상 연산 됩니다.

 

 

 

ex) tran_gan_window_X라는 데이터의 타입을 변형할시 코드

train_gan_window_X = torch.tensor([train_gan_window_X], dtype = torch.float32)