콜백 만들기 : 가장 좋은 모델을 자동 저장하도록 함. 로그도 저장하도록 함.
from tensorflow.keras.callbacks import ModelCheckpoint
cp = ModelCheckpoint(CHECKPOINT_PATH, monitor = 'val_accuracy', save_best_only = True, verbose = 1)
from tensorflow.keras.callbacks import CSVLogger
csv_logger = CSVLogger(LOGFILE_PATH, append = True)callbacks 파라미터에 배열로 객체를 넣어주면 콜백함수 적용이 완료된다.
그러면 학습시 지정한경로로 모델과 로그가 저장된다.
epoch_history = model.fit(train_generator , epochs = 40, validation_data = (X_val, y_val), callbacks = [cp, csv_logger], steps_per_epoch = 10)'인공지능 > 텐서플로우' 카테고리의 다른 글
| Fine Tuning을 하기위한 코드와 설명 (0) | 2022.06.16 |
|---|---|
| Transfer Learning을 하기 위한 코드와 설명 (0) | 2022.06.16 |
| 파일을, training과 test 디렉토리로 나눠서 저장하는 방법 (0) | 2022.06.15 |
| ImageDataGenerator를 이용해서 데이터 증강 하는 방법 (0) | 2022.06.15 |
| 이미지 파일을, 학습데이터로 만드는 방법 (텐서플로우의 ImageDataGenerator) (0) | 2022.06.15 |