Ví dụ gọi lại Keras để lưu mô hình sau mỗi kỷ nguyên?

Ai đó có thể xin vui lòng gửi một ví dụ đơn giản về Keras bằng cách sử dụng một cuộc gọi lại để lưu một mô hình sau mỗi kỷ nguyên không? Tôi có thể tìm thấy các ví dụ về việc tiết kiệm trọng lượng, nhưng tôi muốn có thể lưu một mô hình hoạt động hoàn toàn sau mỗi kỷ nguyên đào tạo.

Đặt 'save_weights_only' thành Sai trong cuộc gọi lại Keras 'ModelCheckpoint' sẽ lưu mô hình đầy đủ; ví dụ này được lấy từ liên kết ở trên sẽ lưu một mô hình đầy đủ mỗi kỷ nguyên, bất kể hiệu suất:

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

Một số ví dụ khác được tìm thấy ở đây , bao gồm chỉ lưu các mô hình được cải thiện và tải các mô hình đã lưu.

Đảm bảo bao gồm biến epoch trong filepath của bạn. Nếu không, mô hình đã lưu của bạn sẽ được thay thế sau mỗi kỷ nguyên.

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

Để biết thêm ví dụ, kiểm tra ở đây .