github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

fix logger for moe

pull/182/head
zhanglei 2023-09-22 14:56:43 +08:00
parent 36d1bd2f41
commit 3df0a51555
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
internlm/initialize/launch.py
View File

@ -365,11 +365,12 @@ def launch(
f"data parallel size: {gpc.data_parallel_size}, pipeline parallel size: {gpc.pipeline_parallel_size}, "
f"tensor parallel size: {gpc.tensor_parallel_size}",
)
logger.info(
f"Creating MoE with num_experts: {gpc.config.model.num_experts} | "
f"expert parallel size: {gpc.expert_parallel_size} | "
f"number of local experts: {gpc.config.model.num_experts//gpc.expert_parallel_size}"
)
if hasattr(gpc.config.model, "num_experts") and gpc.config.model.num_experts > 1:
logger.info(
f"Creating MoE with num_experts: {gpc.config.model.num_experts} | "
f"expert parallel size: {gpc.expert_parallel_size} | "
f"number of local experts: {gpc.config.model.num_experts//gpc.expert_parallel_size}"
)
def launch_from_slurm(