From 111bde9725cf294abc66ba26499ec1413cb7ec70 Mon Sep 17 00:00:00 2001 From: qiaolongfei Date: Tue, 7 Aug 2018 15:44:13 +0800 Subject: [PATCH] add warning to dist lookup table optimizer --- python/paddle/fluid/transpiler/distribute_transpiler.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index 2dc0bda12eb..b0a100e1db3 100644 --- a/python/paddle/fluid/transpiler/distribute_transpiler.py +++ b/python/paddle/fluid/transpiler/distribute_transpiler.py @@ -936,6 +936,10 @@ class DistributeTranspiler(object): } outputs = {"ParamOut": [param_var]} # only support sgd now + import logging + logging.warn( + "distribute lookup table only support sgd optimizer, change it's optimizer to sgd instead of " + + table_opt_op.type) table_opt_block.append_op(type="sgd", inputs=inputs, outputs=outputs) # add table parameter gradient and it's block id to grad_to_block_id -- GitLab