fix dataloader
add retraining model