songsenand
/
SUInput
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

调整数据加载器工作进程数以提高训练效率

This commit is contained in:
songsenand 2026-02-09 01:18:17 +08:00
parent f2c260de72
commit f6c58ab4c6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/suinput/dataset.py
View File

@ -595,7 +595,7 @@ if __name__ == "__main__":
dataloader = DataLoader( dataloader = DataLoader(
dataset, dataset,
batch_size=256, batch_size=256,
num_workers=12, num_workers=15,
worker_init_fn=worker_init_fn, worker_init_fn=worker_init_fn,
# pin_memory=True, # pin_memory=True,
collate_fn=custom_collate, collate_fn=custom_collate,