2330 samples Done in 361s Saving the dataset (0/1 shards): 0%| | 0/2330 [00:00