improve the training framework