RuntimeError: «nll_loss_forward_reduce_cuda_kernel_2d_index» не реализовано для «Int»: Pytorch ⇐ Python

[Anonymous]

Итак, я пытался написать чат-бота с помощью Pytorch, следуя этому руководству.
Код: (минимальный, воспроизводимый)

Код: Выделить всё

tags = []
for intent in intents['intents']:
tag = intent['tag']
tags.append(tag)

tags = sorted(set(tags))

X_train = []
X_train = np.array(X_train)

class ChatDataset(Dataset):
def __init__(self):
self.n_sample = len(X_train)
self.x_data = X_train

#Hyperparameter
batch_size = 8
hidden_size = 47
output_size = len(tags)
input_size = len(X_train[0])
learning_rate = 0.001
num_epochs = 1000

dataset = ChatDataset()
train_loader = DataLoader(dataset=dataset, batch_size=batch_size, shuffle=True, num_workers=0)

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') # using gpu
model = NeuralNet(input_size, hidden_size, output_size).to(device)

# loss and optimizer
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate)

for epoch in range(num_epochs):
for (words, labels) in train_loader:
words = words.to(device)
labels = labels.to(device)

#forward
outputs = model(words)
loss = criterion(outputs, labels) #the line where it is showing the problem

#backward and optimizer step
optimizer.zero_grad()
loss.backward()
optimizer.step()

if (epoch +1) % 100 == 0:
print(f'epoch {epoch+1}/{num_epochs}, loss={loss.item():.4f}')

print(f'final loss, loss={loss.item():.4f}')

Полный код (при необходимости)
Я получаю эту ошибку при попытке получить функцию потерь.

Код: Выделить всё

RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'

Обратная связь:

Код: Выделить всё

Traceback (most recent call last): File "train.py", line 91, in  loss = criterion(outputs, labels) File "C:\Users\PC\anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1102, in _call_impl return forward_call(*input, **kwargs) File "C:\Users\PC\anaconda3\lib\site-packages\torch\nn\modules\loss.py", line 1150, in forward return F.cross_entropy(input, target, weight=self.weight, File "C:\Users\PC\anaconda3\lib\site-packages\torch\nn\functional.py", line 2846, in cross_entropy return torch._C._nn.cross_entropy_loss(input, target, weight, _Reduction.get_enum(reduction), ignore_index, label_smoothing) RuntimeError: "nll_loss_forward_reduce_cuda_kernel_2d_index" not implemented for 'Int'

Но, глядя на руководство, кажется, что там оно работает идеально, тогда как в моем случае это не так.
Что теперь делать?
Спасибо.>

Подробнее здесь: https://stackoverflow.com/questions/697 ... mented-for