Bug with calculating the Training Loss

[Saleh Ashkboos]

Federated-Learning-PyTorch/src/federated_main.py

Line 105 in 235b8f0

 idxs=user_groups[idx], logger=logger) 

Here, I think we have to replace the idx with c because we want to calculate the Training error of the global_model on all training data (after the averaging).

[heymyNLI]

I agree with sashkboos

[Bigmoumou]

Agree. It should loop all the clients.

[flash]

of couse, it is a mistake apparently