add output_layer->train_batch after all pre trained