diff --git a/python/paddle/fluid/transpiler/distribute_transpiler.py b/python/paddle/fluid/transpiler/distribute_transpiler.py index 121c36e477327d4d0e7b1cba1713e68ce4d06e03..b0e9a6599730cde98338edc3e12eb81fc8f793c3 100644 --- a/python/paddle/fluid/transpiler/distribute_transpiler.py +++ b/python/paddle/fluid/transpiler/distribute_transpiler.py @@ -31,6 +31,7 @@ Steps to transpile pserver: from __future__ import print_function import math +import random import numpy as np from ps_dispatcher import RoundRobin, HashName, PSDispatcher @@ -197,7 +198,8 @@ class DistributeTranspiler(object): # shuffle the map will avoid the uneven distribution above grad_var_mapping_items = self.grad_var_mapping.items() if not slice_var_up: - np.random.shuffle(grad_var_mapping_items) + random.seed(self.trainer_num) + random.shuffle(grad_var_mapping_items) for orig_varname, splited_vars in grad_var_mapping_items: eplist = ps_dispatcher.dispatch(splited_vars)