Skip to content

Commit 111bde9

Browse files
committed
add warning to dist lookup table optimizer
1 parent b2f2887 commit 111bde9

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

python/paddle/fluid/transpiler/distribute_transpiler.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -936,6 +936,10 @@ def _create_table_optimize_block(self, pserver_index, pserver_program,
936936
}
937937
outputs = {"ParamOut": [param_var]}
938938
# only support sgd now
939+
import logging
940+
logging.warn(
941+
"distribute lookup table only support sgd optimizer, change it's optimizer to sgd instead of "
942+
+ table_opt_op.type)
939943
table_opt_block.append_op(type="sgd", inputs=inputs, outputs=outputs)
940944

941945
# add table parameter gradient and it's block id to grad_to_block_id

0 commit comments

Comments
 (0)