RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

torchsummary로 pytorch의 모델의 구조를 확인하려할때 위와 같은 에러가 발생하였다.

모델이나 data가 GPU에 올라가있지 않아서 발생하는 에러라고 한다.

# dataset을 GPU에 올릴 때
torch.tensor(..., device="cuda") 
torch.tensor(...).cuda() 
torch.tensor(...).to("cuda")

# model을 GPU에 올릴 때
model.to("cuda")