Function on_state_reset

test/integration/data/elastic_tensorflow_keras_main.py:108–109

()

def on_state_reset():
    tf.keras.backend.set_value(model.optimizer.lr, lr * hvd.size())


state = hvd.elastic.KerasState(model, batch=0, epoch=0, commits=0, rendezvous=0)