Ps Strategy

PS StrategyBuilder.

class PS(local_proxy_variable=False, sync=True, staleness=0)

Bases: autodist.strategy.base.StrategyBuilder

PS StrategyBuilder.

Generates a Strategy that synchronizes every variable using Parameter Servers. Each variable is only assigned to one Parameter Server.

build(graph_item, resource_spec)

Build PS strategy.