Переменная не преобразуется в Cuda - PullRequest
0 голосов
/ 01 июля 2019

Я не могу получить свои входные данные для преобразования в 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'
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...