@Skanda_Vyas , the message says there are some invalid files in your dataset. You can just browse your dataset and see if the files are really empty. In fact, this is done in the pre-processing step where you will remove any invalid data before you run the split process.
@Skanda_Vyas The numbers don’t line up. Please see the expected output and fix your code.
As far as removing invalid images are concerned, don’t remove them. Ignore and print the expected message from within the function. The grader might test that as well.