-
Notifications
You must be signed in to change notification settings - Fork 4
Open
Description
def _state(self, label_job_id, role="worker"): # whether this action selection leads to worker increment or ps increment # cluster_state = self.cluster.get_cluster_state() input = self.observe() # NN input label = np.zeros(pm.ACTION_DIM) for i in range(pm.SCHED_WINDOW_SIZE): job = self.window_jobs[i] if job and job.id == label_job_id: if pm.PS_WORKER: if pm.BUNDLE_ACTION: if role == "worker": label[i * 3] = 1 elif role == "ps": label[i * 3 + 1] = 1 elif role == "bundle": label[i * 3 + 2] = 1 else: if role == "worker": label[i * 2] = 1 elif role == "ps": label[i * 2 + 1] = 1 else: label[i] = 1 self.data.append((input,label))
Does anybody know the meaning of label?
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels