Как сохранить импульс во время обучения в Keras?

Question

Я хочу, чтобы сохранить состояние обучения моей нейронной сети с помощью SGD. Я знаю, что могу сохранить весы, используя функцию save_weights. Я не знаю, как сохранить импульс градиента приличный. Кто-нибудь знает как это сделать?

Answer 1

model.optimizer.get_state() и model.optimizer.set_state() - см. https://github.com/fchollet/keras/issues/454.