API名称 |
是否支持 |
限制与说明 |
|---|---|---|
torch.distributed.checkpoint.load_state_dict |
否 |
|
torch.distributed.checkpoint.save_state_dict |
否 |
|
torch.distributed.checkpoint.StorageReader |
否 |
|
torch.distributed.checkpoint.StorageWriter |
否 |
|
torch.distributed.checkpoint.LoadPlanner |
否 |
|
torch.distributed.checkpoint.LoadPlan |
否 |
|
torch.distributed.checkpoint.ReadItem |
否 |
|
torch.distributed.checkpoint.SavePlanner |
否 |
|
torch.distributed.checkpoint.SavePlan |
否 |
|
torch.distributed.checkpoint.WriteItem |
否 |
|
torch.distributed.checkpoint.FileSystemReader |
否 |
|
torch.distributed.checkpoint.FileSystemWriter |
否 |
|
torch.distributed.checkpoint.DefaultSavePlanner |
否 |
|
torch.distributed.checkpoint.DefaultSavePlanner.lookup_object |
否 |
|
torch.distributed.checkpoint.DefaultSavePlanner.transform_object |
否 |
|
torch.distributed.checkpoint.DefaultLoadPlanner |
否 |
|
torch.distributed.checkpoint.DefaultLoadPlanner.lookup_tensor |
否 |
|
torch.distributed.checkpoint.DefaultLoadPlanner.transform_tensor |
否 |