Tensor에도 type이 있다

요약:
Expected a 'cuda' device type for generator but found 'cpu'
오류시
torch.set_default_tensor_type 부분을 지운다!

---

Expected a 'cuda' device type for generator but found 'cpu'
라는 오류를 만나게 되었다.

그 이유는

if torch.cuda.is_available():
    if args.cuda:
        torch.set_default_tensor_type('torch.cuda.FloatTensor')
    if not args.cuda:
        print("WARNING: It looks like you have a CUDA device, but aren't " +
              "using CUDA.\nRun with --cuda for optimal training speed.")
        torch.set_default_tensor_type('torch.FloatTensor')
else:
    torch.set_default_tensor_type('torch.FloatTensor')

이라고 선언이 되어있었고 cuda를 사용하기 때문에 자동으로 tensor type이 torch.cuda.FloatTensor로 되어있었기 때문

하지만 다른 코드에서 .cpu()로 tesnor의 형을 변환했기 때문에 이를 넘기는 과정에서 오류가 생겼었다.