Loading dataset from disk:   0%|                                                                                                                                                        | 0/181 [00:00<?, ?it/s]
Loading dataset from disk: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 181/181 [00:00<00:00, 14942.80it/s]

Loading checkpoint shards:   0%|                                                                                                                                                          | 0/4 [00:00<?, ?it/s]
Loading checkpoint shards:  50%|█████████████████████████████████████████████████████████████████████████                                                                         | 2/4 [00:13<00:13,  6.51s/it]
Loading checkpoint shards:  75%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████▌                                    | 3/4 [00:23<00:08,  8.34s/it]
Loading checkpoint shards: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 4/4 [00:24<00:00,  5.58s/it]
Loading checkpoint shards: 100%|██████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 4/4 [00:24<00:00,  6.22s/it]