② random_flip_left_right는 90도 회전한 이미지를 생성합니다.
▲ 그림 6-4 random_flip_left_right
모델을 만든 후에는 파라미터를 훈련시켜야 합니다. 주어진 에포크(epoch)1 수만큼 모델을 학습시켜 봅시다. 또한, tf.keras.callbacks.TensorBoard 콜백을 추가하여 로그가 생성되고 저장되도록 합시다.
코드 6-6 텐서보드에서 모델 훈련 과정 살펴보기
log_dir = "../chap6/img/log6-1/"
tensorboard_callback = tf.keras.callbacks.TensorBoard(log_dir=log_dir,
histogram_freq=1, profile_batch=0) ------ ①
model.fit(train_generator,
epochs=EPOCHS,
steps_per_epoch=train_num // BATCH_SIZE,
validation_data=valid_generator,
validation_steps=valid_num // BATCH_SIZE,
callbacks=[tensorboard_callback],
verbose=1) ------ ②
1 알고리즘이 전체 훈련 데이터셋을 반복해서 학습하는 횟수입니다.