修改Dataloader提升训练效率

2025-08-07 11:00:36 +08:00
parent 27ffb62223
commit bc896fc688
1 changed files with 1 additions and 0 deletions
--- a/train_compare.py
+++ b/train_compare.py
@ -337,6 +337,7 @@ def run_training(rank, world_size, conf):
        pin_memory=conf['base']['pin_memory'],
        drop_last=False
    )
+
    # 构建组件字典
    components = {
        'conf': conf,