I am trying to use a MobileNet model but facing above mentioned issue . I don\'t know if it is occuring due to train_test_split or else . Architecture is shown below Can I use m