모델이 epoch를 도는 순간에는 model.save를 사용할 순 없습니다.
이럴때 사용할 수 있게 callback이라는 기능을 제공을 해주었습니다.
정한 epoch 또는 1 epoch이 끝날때마다 이 함수를 불러서 정해놓은 기능을 호출하는데요.
이 글에서는 모델 저장하는 법을 알려드리겠습니다.
1. from keras.callbacks import ModelCheckpoint
2.
다른 param은 Keras.io를 참고하세요. weight만 저장하는 기능도 있어 보입니다.
https://keras.io/callbacks/#modelcheckpoint
3.
이런식으로 callbacks에 세팅을 해주시면 자동으로 정해놓은 path에 저장을 해줍니다.
4.
'# Machine Learning > Keras Implementation' 카테고리의 다른 글
keras Custom generator - 1 (1) | 2019.07.29 |
---|---|
TTA(test time augmentation) with 케라스 (2) | 2019.07.01 |
Keras ImageDataGenerator flow 사용 (0) | 2018.12.20 |
CNN visualization code (3) (0) | 2018.11.26 |
CNN visualization code (2) (0) | 2018.11.26 |