应用 Zero-3 后,LoRA 显示为空 lora 权重 [0]

作者: jiangxinke创建于 2025年4月16日更新于 2025年5月11日

Before using zero 3: base_model.model.model.layers.20.self_attn.q_proj.lora_A.default.weight: torch.Size([8, 1536]) base_model.model.model.layers.20.self_attn.q_proj.lora_B.default.weight: torch.Size([1536, 8]) base_model.model.model.layers.20.self_attn.v_proj.lora_A.default.weight: torch.Size([8, 1536]) base_model.model.model.layers.20.self_attn.v_proj.lora_B.default.weight: torch.Size([256, 8]) base_model.model.model.layers.21.self_attn.q_proj.lora_A.default.weight: torch.Size([8, 1536]) base_model.model.model.layers.21.self_attn.q_proj.lora_B.default.weight: torch.Size([1536, 8]) base_model.model.model.layers.21.self_attn.v_proj.lora_A.default.weight: torch.Size([8, 1536]) After using zero 3: module.base_model.model.model.layers.21.self_attn.q_proj.lora_A.default.weight: torch.Size([0]) ...

内容来源: deepspeedai/DeepSpeedExamples