Bug with calculating the Training Loss

[sashkboos]

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.

[wangclin]

of couse, it is a mistake apparently