change to train_batch(batch,batch_size)