From 93034b192acf17b8799e14cd57888613104ab51f Mon Sep 17 00:00:00 2001 From: GuyPerets106 Date: Sat, 3 Aug 2024 17:48:09 +0000 Subject: [PATCH] [TORCH] FED DEBUG --- .../NerlnetApp/src/Bridge/onnWorkers/workerFederatedClient.erl | 1 + 1 file changed, 1 insertion(+) diff --git a/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerFederatedClient.erl b/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerFederatedClient.erl index df9b51be..256ecfa0 100644 --- a/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerFederatedClient.erl +++ b/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerFederatedClient.erl @@ -173,6 +173,7 @@ post_train({GenWorkerEts, _Data}) -> SyncCount = ets:lookup_element(FedClientEts, sync_count, ?ETS_KEYVAL_VAL_IDX), MaxSyncCount = ets:lookup_element(FedClientEts, sync_max_count, ?ETS_KEYVAL_VAL_IDX), if SyncCount == MaxSyncCount -> + io:format("~p sent averaging request to server~n", [MyName]), ModelID = ets:lookup_element(GenWorkerEts, model_id, ?ETS_KEYVAL_VAL_IDX), WeightsTensor = nerlNIF:call_to_get_weights(ModelID), ServerName = ets:lookup_element(FedClientEts, server_name, ?ETS_KEYVAL_VAL_IDX),