to train student network