def METHOD_NAME(self, enable_training=False):
    return self.client_module.fc1.weight, \
           self.client_module.fc1.bias, \
           self.client_module.fc2.weight, \
           self.client_module.fc2.bias