each dataset is now saved in a single torch dump file