dataloader succeed 1

test on the whole train_set