From c2cb8f0033ebd0cbfc6e9f94b233355a853ac242 Mon Sep 17 00:00:00 2001 From: Li Fuchen Date: Mon, 24 Aug 2020 11:00:18 +0800 Subject: [PATCH] fix docstr of ctc_loss and CTCLoss, test=document_fix (#26584) --- python/paddle/nn/functional/loss.py | 8 ++++---- python/paddle/nn/layer/loss.py | 8 ++++---- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/python/paddle/nn/functional/loss.py b/python/paddle/nn/functional/loss.py index ba057f38bb..68d0f39345 100644 --- a/python/paddle/nn/functional/loss.py +++ b/python/paddle/nn/functional/loss.py @@ -1007,10 +1007,10 @@ def ctc_loss(log_probs, label_lengths = np.array([3, 3]).astype("int64") paddle.disable_static() - log_probs = paddle.to_variable(log_probs) - labels = paddle.to_variable(labels) - input_lengths = paddle.to_variable(input_lengths) - label_lengths = paddle.to_variable(label_lengths) + log_probs = paddle.to_tensor(log_probs) + labels = paddle.to_tensor(labels) + input_lengths = paddle.to_tensor(input_lengths) + label_lengths = paddle.to_tensor(label_lengths) loss = F.ctc_loss(log_probs, labels, input_lengths, diff --git a/python/paddle/nn/layer/loss.py b/python/paddle/nn/layer/loss.py index 6ce036f41f..21cd8535a1 100644 --- a/python/paddle/nn/layer/loss.py +++ b/python/paddle/nn/layer/loss.py @@ -840,10 +840,10 @@ class CTCLoss(fluid.dygraph.Layer): label_lengths = np.array([3, 3]).astype("int64") paddle.disable_static() - log_probs = paddle.to_variable(log_probs) - labels = paddle.to_variable(labels) - input_lengths = paddle.to_variable(input_lengths) - label_lengths = paddle.to_variable(label_lengths) + log_probs = paddle.to_tensor(log_probs) + labels = paddle.to_tensor(labels) + input_lengths = paddle.to_tensor(input_lengths) + label_lengths = paddle.to_tensor(label_lengths) loss = paddle.nn.CTCLoss(blank=0, reduction='none')(log_probs, labels, input_lengths, -- GitLab