본문 바로가기

Deep Learning

fit() vs fit_generator in keras

Keras에서 학습 시 fit()과 fit_generator() 차이

fit()

전체 data set을 한번에 fit method를 통과 시킴

즉, 전체 data set을 메모리에 올릴 수 있어야 사용할 수 있음

 

fit_generator()

generator를 통해 데이터를 load

generator는 multiprocessing을 진행 할 시 데이터 중복을 방지

메모리에 한번에 올리지 못할 때 사용

'Deep Learning' 카테고리의 다른 글

코드 재 사용에 관한 라이센스  (1) 2024.09.07
coco 데이터셋 license  (0) 2024.09.06
CONNX  (0) 2022.10.04
ONNX  (0) 2022.08.31
TensorRT  (0) 2020.08.26