torch.from_numpy expects a single numpy array in the data types mentioned in the error message.
Your dataset.npy seems to contain an array with different objects in it.
I’m not sure if they only have a different shape or also differ in other attributes, but you would have to make sure to create a single array out of your data before passing it to torch.from_numpy.