Keras: ModelCheckpoint сохранить в переменную вместо файла?

Question

Keras: ModelCheckpoint сохранить в переменную вместо файла?

1

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

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

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

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

Eskalior 16 янв. 2019, в 10:24

Источник

1

Вы всегда можете создать свою собственную реализацию обратного вызова, которая делает то, что вы хотите.
Matias Valdenegro 16 янв. 2019, в 08:59
0

Это правда. Если нет другого решения, это может быть то, что я буду делать. Благодарю.
Eskalior 16 янв. 2019, в 09:03

Теги:

python

keras

neural-network

1 ответ

Ещё вопросы

Вы всегда можете создать свою собственную реализацию обратного вызова, которая делает то, что вы хотите.
Это правда. Если нет другого решения, это может быть то, что я буду делать. Благодарю.

BlackBear · Answer 1 · 2019-01-16T07-17-00.000Z

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

Я попытался передать объект файла h5py в качестве параметра, но обратный вызов modelCheckpoint требует только строку пути к файлу в качестве параметра: github.com/keras-team/keras/blob/master/keras/callbacks.py
Вы можете написать свой обратный звонок, это не будет много работы
Я реализовал индивидуальное решение сейчас, и оно работает. Хотя не так много времени, как я ожидал. Похоже, что процесс сохранения был уже относительно быстрым.