A question about train_one_step() method.

[sshpark]

pFedMe/FLAlgorithms/users/userperavg.py

Line 66 in 96863e0

X, y = self.get_next_test_batch()

Hi CharlieDinh, I' m Sorry to take up your time. Can you tell me why self.get_next_test_batch() is used to do the update in the first step and then self.get_next_train_batch() is used to do the update in the second step? Can I do both updates with self.get_next_train_batch() ?

[hesetone]

pFedMe/FLAlgorithms/users/userperavg.py

Line 66 in 96863e0

X, y = self.get_next_test_batch()

Hi CharlieDinh, I' m Sorry to take up your time. Can you tell me why self.get_next_test_batch() is used to do the update in the first step and then self.get_next_train_batch() is used to do the update in the second step? Can I do both updates with self.get_next_train_batch() ?

se se fa dou, orz, ......

[CharlieDinh]

Hi Sshpark, Sorry for the late reply. I've seen your comment yet.
Actually, PerFedAvg �requires a one-step update using the test data. So both should be get_next_test_batch. I updated the code.

[sshpark]

Hi Sshpark, Sorry for the late reply. I've seen your comment yet.
Actually, PerFedAvg �requires a one-step update using the test data. So both should be get_next_test_batch. I updated the code.

Thank you very much for your answer!