모델을 훈련시키려고 train_GAN을 호출합니다. train_dataset 입력을 사용하여 30 에포크만큼 학습시킵니다.

코드 13-26 모델 훈련 및 생성 이미지 출력

train_GAN(train_dataset, 30)

다음은 모델을 훈련시킨 후 이미지를 생성한 결과입니다.

에포크 1 은/는 72.8335235118866 초
에포크 2 은/는 59.848169565200806 초
에포크 3 은/는 60.672967195510864 초
...(중간 생략)...
에포크 28 은/는 63.24000358581543 초
에포크 29 은/는 61.53556036949158 초
에포크 30 은/는 61.84970951080322 초

▲ 그림 13-16 GAN 훈련 출력 결과

에포크 수가 증가할수록 진짜 이미지와 유사한 결과가 출력되는 것을 확인할 수 있습니다. 더 많은 에포크를 사용한다면 더욱 진짜 같은 이미지를 생성할 수 있을 것입니다.