[TORCH] FED DEBUG

leondavi · Aug 2, 2024 · e794774 · e794774
1 parent 18b76f5
commit e794774
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerGeneric.erl b/src_erl/NerlnetApp/src/Bridge/onnWorkers/workerGeneric.erl
@@ -367,6 +367,7 @@ update_client_avilable_worker(MyName) ->
 stream_handler(StreamPhase , ModelPhase , StreamName , DistributedBehaviorFunc) -> 
   GenWorkerEts = get(generic_worker_ets),
   MyName = ets:lookup_element(GenWorkerEts, worker_name, ?ETS_KEYVAL_VAL_IDX),
+  io:format("~p got ~p from ~p~n",[MyName, StreamPhase, StreamName]),
   % io:format("~p got ~p from ~p~n",[MyName, StreamPhase, StreamName]),
   ClientPid = ets:lookup_element(GenWorkerEts, client_pid, ?ETS_KEYVAL_VAL_IDX),
   ActiveStreams = ets:lookup_element(GenWorkerEts, active_streams, ?ETS_KEYVAL_VAL_IDX),

diff --git a/src_erl/NerlnetApp/src/Client/clientStatem.erl b/src_erl/NerlnetApp/src/Client/clientStatem.erl
@@ -293,8 +293,9 @@ training(cast, In = {stream_ended , Pair}, State = #client_statem_state{etsRef =
   ListOfActiveWorkersSources = ets:lookup_element(EtsRef, active_workers_streams, ?DATA_IDX),
   UpdatedListOfActiveWorkersSources = ListOfActiveWorkersSources -- [Pair],
   ets:update_element(EtsRef, active_workers_streams, {?DATA_IDX, UpdatedListOfActiveWorkersSources}),
+  io:format("@Client: ~p~n",[UpdatedListOfActiveWorkersSources]),
   case length(UpdatedListOfActiveWorkersSources) of 
-    0 ->  ets:update_element(EtsRef, all_workers_done, {?DATA_IDX, true});
+    0 ->  io:format("GOT HERE~n"), ets:update_element(EtsRef, all_workers_done, {?DATA_IDX, true});
     _ ->  ok
   end,
   {next_state, training, State#client_statem_state{etsRef = EtsRef}};