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

revert-24895-update_cub
liym27 5 years ago committed by GitHub
parent fd66d76231
commit ac2010f4b6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -31,7 +31,8 @@ IMAGENET1000 = 1281167
base_lr = 0.001 base_lr = 0.001
momentum_rate = 0.9 momentum_rate = 0.9
l2_decay = 1e-4 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 epoch_num = 1
place = fluid.CUDAPlace(0) if fluid.is_compiled_with_cuda() \ place = fluid.CUDAPlace(0) if fluid.is_compiled_with_cuda() \
else fluid.CPUPlace() else fluid.CPUPlace()

Loading…
Cancel
Save