How to know the amount of data transported in simulation?

[poormidfielder]

Hi, recently I'm focusing on the transportation in FedML simulation. I want to know that if there is any method to know the amount of data transported? For example, in a simulation of sp-fedavg-cnn, there should be a process of sending data when aggregating and a process of receiving data when applying new model parameters. So how could I know the amount of data transported in these processes?

[chaoyanghe]

@Amir-zsh Hi Amir, I remember you have developed such APIs to count the communication cost. Could you please help to wrapper it as APIs?

[fedml-dimitris]

@poormidfielder We will revisit this soon.