YAO/YAO-scheduler
1
0
Fork 0
mirror of https://github.com/newnius/YAO-scheduler.git synced 2025-06-07 14:21:55 +00:00
Code Issues Releases Wiki Activity

update

Browse Source
This commit is contained in:
Newnius 2020-06-26 12:57:41 +08:00
parent b75488623f
commit f17b594744
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/optimizer.go
View File

@ -107,6 +107,11 @@ func (optimizer *Optimizer) feedStats(job Job, role string, stats [][]TaskStatus
}
params["ps_number"] = psNumber
params["worker_number"] = workerNumber
if role == "PS" {
params["role"] = 1
} else {
params["role"] = 0
}
exceptions := map[string]bool{}
exceptions["train_dir"] = true
@ -401,9 +406,6 @@ func (optimizer *Optimizer) PredictReq(job Job, role string) MsgJobReq {
cmd := ""
params := map[string]int{}
log.Info(job)
log.Info(role)
psNumber := 0
workerNumber := 0
flag := false
@ -421,6 +423,11 @@ func (optimizer *Optimizer) PredictReq(job Job, role string) MsgJobReq {
}
params["ps_number"] = psNumber
params["worker_number"] = workerNumber
if role == "PS" {
params["role"] = 1
} else {
params["role"] = 0
}
if !flag {
return res
}