Keras: ModelCheckpoint сохранить в переменную вместо файла? - PullRequest
0 голосов
/ 16 января 2019

Я хочу обучить мою модель для эпох X и получить лучший результат с точки зрения минимальной ошибки проверки.

Решение, которое я нашел, состояло в том, чтобы использовать обратный вызов ModelCheckpoint, который можно настроить для переопределения сохранения модели, если результат текущей эпохи был лучше.

Однако в худшем случае это означает, что в каждую эпоху модель записывается на жесткий диск, что приводит к огромным затратам времени.

Есть ли возможность сохранить модель в переменную, поэтому мне нужно сохранить ее только один раз? Или я могу сделать еще одну оптимизацию?

1 Ответ

0 голосов
/ 16 января 2019

Согласно строке документации save_model , путь также может быть объектом h5py.File, и в документации по h5py содержится пример файлов в памяти .

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...