Я не могу получить свои входные данные для преобразования в cuda - не знаю почему. Я дважды проверил, что мое устройство = CUDA и не могу получить эту переменную для преобразования.
Я пропускаю какой-то способ начать куда? Я думал, что синтаксис был .to (устройство), и он не распознает .cuda
Сначала я преобразовал их в pytorch, но затем lables.to (устройство) запутается в виде списка.
print_every = 40
steps = 0
for e in range (epochs):
running_loss = 0
for inputs, labels in enumerate(trainloader):
steps += 1
inputs, labels = inputs.to(device), labels.to(device)
optimizer.zero_grad()
output = model.forward(inputs)
loss = criterion(output, labels)
loss.backward()
optimizer.step()
running_loss += loss.item()
if steps % print_every == 0 :
model.eval()
with torch.no_grad():
validate_loss, accuracy = validate(model, criterion, validloader)
print("Epoch: {}/{}... ".format(e+1, epochs),
"Loss: {:.4f}".format(running_loss/print_every),
"Validation Lost {:.4f}".format(vlost),
"Accuracy: {:.4f}".format(accuracy))
running_loss = 0
model.train()
---------------------------------------------------------------------------
AttributeError Traceback (most recent call last)
<ipython-input-192-47df4e624935> in <module>()
11
12
---> 13 inputs, labels = inputs.to(device), labels.to(device)
14
15 optimizer.zero_grad()
AttributeError: 'int' object has no attribute 'to'