def __init__(self, initial_weight_path, weight_save_path_tmpl, initial_t=0): self.clients = [] self.t = initial_t self.weight_save_path_tmpl = weight_save_path_tmpl self.dataset_loader = DatasetLoader.init_by_setting(dist_params) self.varserver_db = varserverapi.open_db() self.reserved_vars_by_t = defaultdict(list)#variable ids used for specific iteration packer = WeightPack(json.load(open(dist_params.WEIGHT_PACK_PARAM_FILE))) initial_weights = packer.unpack(open(initial_weight_path, "rb").read()) self.optimizer = Optimizer(initial_weights, packer, lr=dist_params.LR) self.weight_id = None self.gradient_ids = [] self.n_gradient_gathered = 0
def __init__(self, initial_weight_path, weight_save_path_tmpl, initial_t=0): self.clients = [] self.t = initial_t self.weight_save_path_tmpl = weight_save_path_tmpl self.dataset_loader = DatasetLoader.init_by_setting(dist_params) self.varserver_db = varserverapi.open_db() self.reserved_vars_by_t = defaultdict( list) #variable ids used for specific iteration self.weights_bin = open(initial_weight_path, "rb").read() self.weight_id = None self.gradient_ids = []