I am trying to run this code with the dataset that you have provided. I was able to process the dataset and convert it to a .npz file. When I try to train it, I get this problem "Loading data should be numpy array and has "ae_out" and "labels" keys." What should I do?