lichen/ieemoo-ai-contrast
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

修改Dataloader提升训练效率

Browse Source
This commit is contained in:
lee
2025-08-07 10:56:32 +08:00
parent ebba07d1ca
commit 27ffb62223
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
train_compare.py
View File

@ -337,7 +337,6 @@ def run_training(rank, world_size, conf):
pin_memory=conf['base']['pin_memory'], pin_memory=conf['base']['pin_memory'],
drop_last=False drop_last=False
) )
# 构建组件字典 # 构建组件字典
components = { components = {
'conf': conf, 'conf': conf,