1
RuntimeError : 입력 유형 (torch.FloatTensor)과 가중치 유형 (torch.cuda.FloatTensor)이 같아야합니다.
다음과 같이 다음 CNN을 훈련하려고하지만 .cuda ()와 관련하여 동일한 오류가 계속 발생하며 수정 방법을 잘 모르겠습니다. 지금까지 내 코드 덩어리가 있습니다. import matplotlib.pyplot as plt import numpy as np import torch from torch import nn from torch import optim import torch.nn.functional as F import torchvision from torchvision import datasets, transforms, …