多机并行计算

2025-08-14 10:09:54 +08:00
parent bc896fc688
commit 99a204ee22
18 changed files with 105 additions and 55 deletions
--- a/tools/similar_analysis.py
+++ b/tools/similar_analysis.py
@ -22,7 +22,7 @@ class SimilarAnalysis:
        """初始化模型和度量方法"""
        tr_tools = trainer_tools(conf)
        backbone_mapping = tr_tools.get_backbone()
-
+        print('model_path {}'.format(conf['models']['model_path']))
        if conf['models']['backbone'] in backbone_mapping:
            model = backbone_mapping[conf['models']['backbone']]()
        else: