diff --git a/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerFederatedServer.erl b/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerFederatedServer.erl index a1b594e8..72013ccc 100644 --- a/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerFederatedServer.erl +++ b/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerFederatedServer.erl @@ -152,17 +152,13 @@ post_train({GenWorkerEts, WeightsTensor}) -> {CurrentModelWeights, BinaryType} = nerlNIF:call_to_get_weights(ModelID), FedServerName = ets:lookup_element(FedServerEts, my_name, ?ETS_KEYVAL_VAL_IDX), AllWorkersWeightsList = TotalWorkersWeights ++ [CurrentModelWeights], - io:format("GOT HERE1~n"), AvgWeightsNerlTensor = generate_avg_weights(AllWorkersWeightsList, BinaryType), - io:format("GOT HERE2~n"), nerlNIF:call_to_set_weights(ModelID, AvgWeightsNerlTensor), %% update self weights to new model - io:format("GOT HERE3~n"), Func = fun(FedClient) -> FedServerName = ets:lookup_element(ThisEts, my_name, ?ETS_KEYVAL_VAL_IDX), W2WPid = ets:lookup_element(ThisEts, w2wcom_pid, ?ETS_KEYVAL_VAL_IDX), w2wCom:send_message_with_event(W2WPid, FedServerName, FedClient, post_train_update, {SyncIdx, AvgWeightsNerlTensor}) end, - io:format("GOT HERE4~n"), WorkersSourcesList = ets:lookup_element(GenWorkerEts, active_streams, ?ETS_KEYVAL_VAL_IDX), WorkersList = [FedWorker || {_MyName, {FedWorker, _Source}} <- WorkersSourcesList], % io:format("Sending new weights to workers ~p~n",[WorkersList]),