I wonder how you did it with the configured code when you changed the dataset to npy file. Looking forward to other people's help and advice.