Hi, I think the file NYU0223_0000 is missing on the data you are providing... It is not in neither train nor test NYU splits for .npz files. Best and thanks for sharing the work.