5fd3dca5f3
* fix batch_size * add test --------- Co-authored-by: yiyixuxu <yixu310@gmail,com>
* fix batch_size * add test --------- Co-authored-by: yiyixuxu <yixu310@gmail,com>