From c657537b4f32ac3a900561b460c2d0de23a723ff Mon Sep 17 00:00:00 2001 From: Xinghai Sun Date: Sun, 3 Sep 2017 00:38:32 +0800 Subject: [PATCH] Correct some typos. --- paddle/operators/dropout_op.cc | 2 +- python/paddle/v2/framework/tests/test_dropout_op.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/paddle/operators/dropout_op.cc b/paddle/operators/dropout_op.cc index 60ad2efbe97..dc416f61942 100644 --- a/paddle/operators/dropout_op.cc +++ b/paddle/operators/dropout_op.cc @@ -62,7 +62,7 @@ class DropoutOpGrad : public framework::OperatorWithKernel { auto mask_dims = ctx.Input("Mask")->dims(); auto out_dims = ctx.Input(framework::GradVarName("Out"))->dims(); PADDLE_ENFORCE_EQ(x_dims, out_dims, - "Dimensions of Input(X) and Out must be the same."); + "Dimensions of Input(X) and Out@Grad must be the same."); PADDLE_ENFORCE_EQ(x_dims, mask_dims, "Dimensions of Input(X) and Mask must be the same."); diff --git a/python/paddle/v2/framework/tests/test_dropout_op.py b/python/paddle/v2/framework/tests/test_dropout_op.py index 3f4738f6145..c5ff55f74c5 100644 --- a/python/paddle/v2/framework/tests/test_dropout_op.py +++ b/python/paddle/v2/framework/tests/test_dropout_op.py @@ -14,7 +14,7 @@ class TestDropoutOpProbZero(unittest.TestCase): self.outputs = {'Out': self.inputs['X'], 'Mask': np.ones((32, 64))} -class TestDropoutOpAllProbOne(unittest.TestCase): +class TestDropoutOpProbOne(unittest.TestCase): __metaclass__ = OpTestMeta def setUp(self): @@ -24,7 +24,7 @@ class TestDropoutOpAllProbOne(unittest.TestCase): self.outputs = {'Out': np.zeros((32, 64)), 'Mask': np.zeros((32, 64))} -class DropoutGradOpTest(GradientChecker): +class TestDropoutGradOp(GradientChecker): def test_dropout_2d(self): op = create_op("dropout") inputs = {'X': np.random.random((10, 5)).astype("float32")} -- GitLab