-
Notifications
You must be signed in to change notification settings - Fork 162
paddle.fleet
Dong Daxiang edited this page May 28, 2020
·
3 revisions
- MPISymetricRoleMaker
- UserDefinedRoleMaker
- UserDefinedCollectiveRoleMaker
- PaddleCloudRoleMaker
- GeneralRoleMaker
import paddle.fleet as fleet
role_maker = PaddleCloudRoleMaker()
fleet.init(role_maker)
collective training
import paddle.fleet as fleet
role_maker = PaddleCloudRoleMaker(collective=True)
fleet.init(role_maker)
class Fleet(object):
def __init__(self, mode):
def is_first_worker(self):
def worker_index(self):
def worker_num(self):
def is_worker(self):
def worker_endpoints(self, to_string=False):
def server_num(self):
def server_index(self):
def server_endpoints(self, to_string=False):
def is_server(self):
def split_files(self, files):
def init(self, role_maker=None):
def all_reduce_worker(self, input, output):
def barrier_worker(self):
def init_worker(self):
def init_server(self, model_dir=None):
def run_server(self):
def stop_worker(self):
def distributed_optimizer(self, optimizer, strategy=None):
def save_inference_model(self,
executor,
dirname,
feeded_var_names,
target_vars,
main_program=None,
export_for_deployment=True):
def save_persistables(self, executor, dirname, main_program=None):