Я может тупой, но раньше у меня была похожая модель:

Question

Я может тупой, но раньше у меня была похожая модель:

y_pred = []
y_true = []
for inputs, labels in tqdm(data_loader, total=len(data_loader)):
with torch.no_grad():
inputs, labels = inputs.cuda(), labels.cuda()
output = bert_punc(inputs)
y_pred += list(output.argmax(dim=1).cpu().data.numpy().flatten())
y_true += list(labels.cpu().data.numpy().flatten())
return y_pred, y_true

Где:
data_set = TensorDataset(torch.from_numpy(X).long(), torch.from_numpy(np.array(y)).long())
data_loader = DataLoader(data_set, batch_size=batch_size, shuffle=shuffle)

И всё летало без проблем. Не знаешь, почему здесь не так?

#database #programming #russian #software

0

07.02.2020

1 ответов

15 просмотров

Max Pakhomov · Accepted Answer

Max Pakhomov

То что из него выходит это inputs, labels в данном случае

0

07.02.2020

215 похожих чатов

Я может тупой, но раньше у меня была похожая модель:

1 ответов

Похожие вопросы