Which is train script, how should I train on different dataset, Do we need to write load data script?