I found the unittest failed due to batch size. Maybe the reason is that our CI machine has limited memory. I decreased the batch size.