def METHOD_NAME(self):
    self.cnt = 0
    # Adapted from https://github.com/Xtra-Computing/NIID-Bench/blob/main/experiments.py#L371
    c_global_para = self.c_global.state_dict()
    c_local_para = self.c_local.state_dict()
    return c_global_para, c_local_para