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

Loading checkpoint shards:   0%|                                                                                                                                                                                                                                                                                   | 0/4 [00:00<?, ?it/s]
Loading checkpoint shards:  25%|██████████████████████████████████████████████████████████████████▊                                                                                                                                                                                                        | 1/4 [00:01<00:03,  1.19s/it]
Loading checkpoint shards:  50%|█████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▌                                                                                                                                     | 2/4 [00:02<00:02,  1.28s/it]
Loading checkpoint shards:  75%|████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████▎                                                                  | 3/4 [00:03<00:01,  1.21s/it]
Loading checkpoint shards: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 4/4 [00:03<00:00,  1.19it/s]
Loading checkpoint shards: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 4/4 [00:03<00:00,  1.01it/s]
Time to load cpu_adam op: 2.235198497772217 seconds