PullFromGitHub/happy-llm
1
0
Fork 0
Code Issues Pull Requests 2 Actions Packages Projects Releases 3 Wiki Activity

refactor: 移除重复的目录创建逻辑

Browse Source
This commit is contained in:
KMnO4-zx
2025-07-25 09:07:17 +08:00
parent f505e8e52c
commit 324b79de91
3 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
docs/chapter5/code/ddp_pretrain.py
View File

@@ -277,7 +277,6 @@ if __name__ == "__main__":
args.save_dir = os.path.join(args.out_dir) # 模型保存目录
# 创建必要的目录
os.makedirs(args.save_dir, exist_ok=True)
os.makedirs(args.out_dir, exist_ok=True)
# 设置随机种子以确保结果可复现

1
docs/chapter5/code/ddp_sft_full.py
View File

@@ -196,7 +196,6 @@ if __name__ == "__main__":
)
max_seq_len = lm_config.max_seq_len
args.save_dir = os.path.join(args.out_dir)
os.makedirs(args.save_dir, exist_ok=True)
os.makedirs(args.out_dir, exist_ok=True)
torch.manual_seed(42)
device_type = "cuda" if "cuda" in args.device else "cpu"

2
docs/chapter5/第五章 动手搭建大模型.md
View File

@@ -1724,7 +1724,6 @@ if __name__ == "__main__":
args.save_dir = os.path.join(args.out_dir) # 模型保存目录
# 创建必要的目录
os.makedirs(args.save_dir, exist_ok=True)
os.makedirs(args.out_dir, exist_ok=True)
# 设置随机种子以确保结果可复现
@@ -1974,7 +1973,6 @@ if __name__ == "__main__":
)
max_seq_len = lm_config.max_seq_len
args.save_dir = os.path.join(args.out_dir)
os.makedirs(args.save_dir, exist_ok=True)
os.makedirs(args.out_dir, exist_ok=True)
torch.manual_seed(42)
device_type = "cuda" if "cuda" in args.device else "cpu"