Explorar o código

add infer code

often hai 5 meses
pai
achega
3098dfdf7f
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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):
         if is_infer:
-            left_vec = get_leftvectors(left_features)
+            left_vec = self.get_leftvectors(left_features)
             return left_vec
         else:
             # 获取两个视频的特征表示