Changes on train/predict/experiment and test_cli files to add an option of multiprocessing on the data loading function