[Fix test_resnet timeout] Reduce batch_size to 2 to avoid unittest timeout.(#26449)

5 years ago · ac2010f4b6
parent fd66d76231
commit ac2010f4b6
1 changed files with 2 additions and 1 deletions
--- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet.py
+++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_resnet.py
@ -31,7 +31,8 @@ IMAGENET1000 = 1281167
 base_lr = 0.001
 momentum_rate = 0.9
 l2_decay = 1e-4
-batch_size = 8
+# NOTE: Reduce batch_size from 8 to 2 to avoid unittest timeout.
+batch_size = 2
 epoch_num = 1
 place = fluid.CUDAPlace(0) if fluid.is_compiled_with_cuda() \
    else fluid.CPUPlace()