def METHOD_NAME(self, parameters, config):
    self.model.set_weights(parameters)
    self.model.METHOD_NAME(
        self.x_train, self.y_train, epochs=1, batch_size=32, verbose=VERBOSE
    )
    return self.model.get_weights(), len(self.x_train), {}