In different settings, how to define train_num and lb_num? Fix normal situation and active learning situation