Kaynağa Gözat

change model dim

often 5 ay önce
ebeveyn
işleme
72abbb3665

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

@@ -115,7 +115,7 @@ class DSSMLayer(nn.Layer):
         left_embedded = self._process_features(left_features, self.left_embeddings)
         paddle.static.Print(left_embedded, message="left_embedded shape:")
         left_vec = left_embedded
-        for layer in self._left_tower:
+        for i, layer in enumerate(self._left_tower):
             left_vec = layer(left_vec)
             paddle.static.Print(left_vec, message=f"After layer {i}:")