From 3d0c8b12a486690a3feb912f6a05809dd89cb679 Mon Sep 17 00:00:00 2001 From: Aurelius84 Date: Fri, 2 Dec 2022 10:06:55 +0800 Subject: [PATCH] [Unitttet] Fix axes error from migrating paddle.squeeze in test_seq2seq (#48620) --- .../unittests/dygraph_to_static/seq2seq_dygraph_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/seq2seq_dygraph_model.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/seq2seq_dygraph_model.py index c9a58b9c78..39a04d7a78 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/seq2seq_dygraph_model.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/seq2seq_dygraph_model.py @@ -297,7 +297,7 @@ class BaseModel(fluid.dygraph.Layer): loss = fluid.layers.softmax_with_cross_entropy( logits=dec_output, label=label, soft_label=False ) - loss = paddle.squeeze(loss, axes=[2]) + loss = paddle.squeeze(loss, axis=[2]) max_tar_seq_len = fluid.layers.shape(tar)[1] tar_mask = fluid.layers.sequence_mask( tar_sequence_length, maxlen=max_tar_seq_len, dtype='float32' @@ -831,7 +831,7 @@ class AttentionModel(fluid.dygraph.Layer): loss = fluid.layers.softmax_with_cross_entropy( logits=dec_output, label=label, soft_label=False ) - loss = paddle.squeeze(loss, axes=[2]) + loss = paddle.squeeze(loss, axis=[2]) max_tar_seq_len = fluid.layers.shape(tar)[1] tar_mask = fluid.layers.sequence_mask( tar_sequence_length, maxlen=max_tar_seq_len, dtype='float32' -- GitLab