Splitting up the images set in training process to avoid the system to load the entire set at once.