diff --git a/train_compare.py b/train_compare.py index 425baf4..b1f8208 100644 --- 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,