add warning to dist lookup table optimizer

revert-12646-feature/jit/xbyak
qiaolongfei 7 years ago
parent b2f28878e1
commit 111bde9725

@ -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

Loading…
Cancel
Save