修改Dataloader提升训练效率

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