Skip to content

完全分片数据并行工具

enable_fsdp_ram_efficient_loading

[[autodoc]] utils.enable_fsdp_ram_efficient_loading

disable_fsdp_ram_efficient_loading

[[autodoc]] utils.disable_fsdp_ram_efficient_loading

merge_fsdp_weights

[[autodoc]] utils.merge_fsdp_weights

FullyShardedDataParallelPlugin

[[autodoc]] utils.FullyShardedDataParallelPlugin