Allocation of 406978560 exceeds 10% of free system memory

현상 : tensorflow 사용 중 "Allocation of 406978560 exceeds 10% of free system memory" 경고와 함께 오류발생

 

내용 : 쥬피터노트북에서 model.fit 중 커널이 죽고 자동으로 시작하는 문제가 발생하여, 파이참에 그대로 옮겨보니 오류를 확인 할 수 있었다.

 

진행 1 : 아래 코드를 추가해보았으나, 경고만 없어질뿐 문제는 똑같았다. (해결안됨)

import os

os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'

 

진행 2 : BATCH_SIZE 변경 (적은값으로)시 정상으로 진행하였음.

model.fit(x_train, y_train, epochs=EPOCHS, batch_size=BATCH_SIZE, callbacks=callbacks, verbose=3)

 

나는 CPU를 이용해서 테스트하고 있어서 해당값을 작게해야 돌아갔음.

BATCH_SIZE는 한번에 신경망에 넣어 학습시킬 양을 뜻함.