dssm train

recommend-model-produce/src/main/python/models/dssm/net.py

@@ -175,11 +175,11 @@ class DSSMLayer(nn.Layer):
         return paddle.concat(embedded_features, axis=1)
 
 
-    def forward(self, vid, left_features, right_features, is_infer=False):
+    def forward(self, left_features, right_features, is_infer=False):
         # paddle.static.Print(left_features, message="left_features shape and value:")
         if is_infer:
             left_vec = self.get_leftvectors(left_features)
-            return vid, left_vec
+            return left_vec
         else:
             # 获取两个视频的特征表示      
             left_vec, right_vec = self.get_vectors(left_features, right_features)

recommend-model-produce/src/main/python/models/dssm/static_model.py

@@ -92,7 +92,7 @@ class StaticModel():
         
         if is_infer:
             vid, left_features = input
-            left_vec = dssm_model(left_features,None,is_infer=True)
+            left_vec = dssm_model(left_features,right_features=None,is_infer=True)
             self.inference_target_var = left_vec
             fetch_dict = {
                 'left_vector': left_vec,