[error] Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

 

 

for i , (inputs, labels)   in enumerate(trainloader, 0):
        inputs, labels = inputs.to(device), labels.to(device)
        optimizer.zero_grad()

        #### error 가 난 위치
        outputs = net(inputs)
        ####
        loss = criterion(outputs, labels)

input은 device("cuda") 로 설정해준반면 모델인 net는 설정해주지 않아 발생한 에러이다.