lin fan 1 年之前
父節點
當前提交
417e08a912
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      video_rank.py

+ 1 - 1
video_rank.py

@@ -970,7 +970,7 @@ def video_sank_pos_rank(data, size, top_K, flow_pool_P, ab_Code='', exp_config=N
     recall_list.append(recall_pos4)
     select_ids = set('')
     recall_num_limit_dict = {}
-    if 'recall_num_limit' in exp_config:
+    if exp_config and 'recall_num_limit' in exp_config:
         recall_num_limit_dict = exp_config['recall_num_limit']
     exp_recall_dict = {}
     for j in range(3):