Browse Source

add infer code

often 5 months ago
parent
commit
3098dfdf7f
1 changed files with 1 additions and 1 deletions
  1. 1 1
      recommend-model-produce/src/main/python/models/dssm/net.py

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

@@ -100,7 +100,7 @@ class DSSMLayer(nn.Layer):
 
 
     def forward(self, left_features, right_features, is_infer=False):
     def forward(self, left_features, right_features, is_infer=False):
         if is_infer:
         if is_infer:
-            left_vec = get_leftvectors(left_features)
+            left_vec = self.get_leftvectors(left_features)
             return left_vec
             return left_vec
         else:
         else:
             # 获取两个视频的特征表示      
             # 获取两个视频的特征表示