train_loss.append(loss.item()). ## evaluation part. model.eval() ... et = torch.tensor(embedding_matrix, dtype=torch.float32).
確定! 回上一頁