func SimpleWorker(queue mpi.MessagesQueue, cmpt *Computable, tau int) {
i := 1
vc := make(map[string]int) // version control map
for {
select {
case data, ok := <-cmpt.Input:
if !ok {
return
}
if i == 1 {
for _, v := range queue.Queues() {
vc[v] = 0
}
}
// ici on va faire des computations
if i%tau == 0 && tau != 1 {
stat := queue.ReadStates(vc)
// Block of code just to covert to the good types
states := make([]models.State, 0, len(stat))
for _, v := range stat {
states = append(states, v.(models.State))
}
// We know append the knew versions
for key, v := range stat {
vc[key] = v.Version()
}
if len(states) != 0 {
acc := models.States(states).Average()
cmpt.Est.Average(acc, data)
} else {
cmpt.Est.Compute(data)
}
} else {
cmpt.Est.Compute(data)
}
queue.Write(cmpt.Name, cmpt.Est.State())
i = i + 1
}
}
}
func Worker(queue mpi.MessagesQueue, cmpt *models.DistributedSGD, tau int) {
i := 1
vc := make(map[string]int) // version control map
var model1 []float64
copy(cmpt.Instance.Model(), model1)
model2 := cmpt.Instance.Model()
for !cmpt.Stop(model1, model2) {
if i == 1 {
for _, v := range queue.Queues() {
vc[v] = 0
}
}
// ici on va faire des computations
if i%tau == 0 && tau != 1 {
stat := queue.ReadStates(vc)
// Block of code just to covert to the good types
states := make([]models.State, 0, len(stat))
for _, v := range stat {
states = append(states, v.(models.State))
}
// We know append the knew versions
for key, v := range stat {
vc[key] = v.Version()
}
if len(states) != 0 {
acc := models.States(states)
cmpt.Average(acc, i)
}
} else {
cmpt.Compute(i)
}
queue.Write(cmpt.Name, cmpt.State())
i = i + 1
}
}