[TORCH] FED DEBUG

leondavi · Aug 3, 2024 · 93034b1 · 93034b1
1 parent 0f15d16
commit 93034b1
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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),