correct unittest

5d2fad7b · LiuChiaChi · 47e375ac · 5d2fad7b
隐藏空白更改
内联并排

Showing with 1 addition and 3 deletion

python/paddle/tests/test_model.py python/paddle/tests/test_model.py +1 -3

未找到文件。
--- a/python/paddle/tests/test_model.py
+++ b/python/paddle/tests/test_model.py
@@ -415,9 +415,7 @@ class TestModelFunction(unittest.TestCase):
        # dynamic saving
        device = paddle.set_device('cpu')
        fluid.enable_dygraph(device)
-        inputs = [InputSpec([None, 20], 'float32', 'x')]
+        model = Model(MyModel(classifier_activation=None))
-        labels = [InputSpec([None, 1], 'int64', 'label')]
-        model = Model(MyModel(classifier_activation=None), inputs, labels)
        optim = fluid.optimizer.SGD(learning_rate=0.001,
                                    parameter_list=model.parameters())
        model.prepare(optimizer=optim, loss=CrossEntropyLoss(reduction="sum"))