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):