Python FullyShardedDataParallel.rekey_optim_state_dict - 2 examples found. These are the top rated real world Python examples of torch.distributed.fsdp.FullyShardedDataParallel.rekey_optim_state_dict extracted from open source projects. You can rate examples to help us improve the quality of examples.