Keras Callback contoh untuk menyimpan model setelah setiap zaman?


17

Dapatkah seseorang tolong posting contoh langsung dari Keras menggunakan panggilan balik untuk menyimpan model setelah setiap zaman? Saya dapat menemukan contoh penghematan bobot, tetapi saya ingin dapat menyimpan model yang berfungsi penuh setelah setiap zaman pelatihan.

Jawaban:


22

Mengatur 'save_weights_only' ke False dalam panggilan balik Keras 'ModelCheckpoint' akan menyimpan model lengkap; contoh ini diambil dari tautan di atas akan menyimpan model lengkap setiap zaman, terlepas dari kinerja:

keras.callbacks.ModelCheckpoint(filepath, monitor='val_loss', verbose=0, save_best_only=False, save_weights_only=False, mode='auto', period=1)

Beberapa contoh lain ditemukan di sini , termasuk hanya menyimpan model yang lebih baik dan memuat model yang disimpan.


10

Pastikan untuk memasukkan variabel zaman dalam filepath Anda. Kalau tidak, model Anda yang disimpan akan diganti setelah setiap zaman.

filepath = "saved-model-{epoch:02d}-{val_acc:.2f}.hdf5"
checkpoint = ModelCheckpoint(filepath, monitor='val_acc', verbose=1, save_best_only=False, mode='max')

Untuk lebih banyak contoh, periksa di sini .


2
Selamat datang di situs ini! Dan terima kasih, saya menghargai jawaban itu
I_Play_With_Data
Dengan menggunakan situs kami, Anda mengakui telah membaca dan memahami Kebijakan Cookie dan Kebijakan Privasi kami.
Licensed under cc by-sa 3.0 with attribution required.