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 = []
