瀏覽代碼

add infer code

often 5 月之前
父節點
當前提交
56bb5f92c9
共有 1 個文件被更改,包括 2 次插入1 次删除
  1. 2 1
      recommend-model-produce/src/main/python/models/dssm/static_model.py

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

@@ -106,7 +106,8 @@ class StaticModel():
             import paddle.distributed.fleet as fleet
             optimizer = fleet.distributed_optimizer(optimizer, strategy)
         if is_infer:
-            optimizer.minimize(paddle.mean(0))
+            zero_var = paddle.zeros(shape=[1], dtype='float32')
+            optimizer.minimize(paddle.mean(zero_var))
         else:
             optimizer.minimize(self._cost)