Browse Source

dssm train

丁云鹏 4 months ago
parent
commit
44dcf15270
1 changed files with 1 additions and 1 deletions
  1. 1 1
      recommend-model-produce/src/main/python/tools/inferv2.py

+ 1 - 1
recommend-model-produce/src/main/python/tools/inferv2.py

@@ -62,7 +62,7 @@ def main():
     # 设置输入
     # 设置输入
     
     
 
 
-    for k,v in input_data2:
+    for k,v in input_data2.items():
         input_handle.copy_from_cpu(v)
         input_handle.copy_from_cpu(v)
         # 运行predictor
         # 运行predictor
         predictor.run()
         predictor.run()