INNER CODE UNIT · Python
set_kungfu_option
tensorlayer/HyperPose · hyperpose/Config/__init__.py:357
def set_kungfu_option(kungfu_option):
'''set the optimizor of parallel training
kungfu distribute training library needs to wrap tensorflow optimizor in
kungfu optimizor, this function is to choose kungfu optimizor wrap type
Parameters
----------
arg1 : Config.KUNGFU
a enum value of enum class Config.KUNGFU
available options:
| Config.KUNGFU.Sync_sgd (SynchronousSGDOptimizer, hyper-parameter-robus)
| Config.KUNGFU.Sync_avg (SynchronousAveragingOptimizer)
| Config.KUNGFU.Pair_avg (PairAveragingOptimizer, communication-efficient)
Returns
-------
None