Loading dataset from disk:   0%|                                                                                                                | 0/64 [00:00<?, ?it/s]
Loading dataset from disk: 100%|████████████████████████████████████████████████████████████████████████████████████████████████████| 64/64 [00:00<00:00, 38774.44it/s]
[2026-03-05 02:08:00,382] [WARNING] [py.warnings._showwarnmsg:109] [PID:2751870] /usr/local/lib/python3.10/dist-packages/torch/distributed/fsdp/fully_sharded_data_parallel.py:678: FutureWarning: FSDP.state_dict_type() and FSDP.set_state_dict_type() are being deprecated. Please use APIs, get_state_dict() and set_state_dict(), which can support different parallelisms, FSDP1, FSDP2, DDP. API doc: https://pytorch.org/docs/stable/distributed.checkpoint.html#torch.distributed.checkpoint.state_dict.get_state_dict .Tutorial: https://pytorch.org/tutorials/recipes/distributed_checkpoint_recipe.html .
  warnings.warn(