학습 저장하고 이어서 하기

Question

5강 Craft_license 파일을 직접 실습해보고 있습니다. 컴퓨터 내장의 GPU를 사용해서 해보고 싶어서 파이참에 GPU설정을 해둔 후에 학습을 시키다가 571번 학습까지 가다가 멈춰서 다시 학습을 이어가고 싶은데 코드를 돌리면 다시 1번부터 반복이 시작됩니다.

파일을 보면 h5파일들도 정상적으로 만들어져 있고

.\detector_carplate.h5 model loaded!

라는 파일을 잘 인식했다는 문구도 나오는데 Epoch이 왜 1부터 시작되는지 모르겠습니다 ㅠㅠ

Answer

안녕하세요~. 우선 답신이 늦어서 죄송합니다ㅠ.

h5 파일을 저장한뒤 정상적으로 불러왔으면 다시 1번 epoch부터 시작하는게 아니라 572번 epoch부터 이어서 정상적으로 진행되고 있는 상태로 예상됩니다.

다만 fit_generator 함수 호출시에 매번 initial_epoch=0 이 0으로 초기화되서 epoch이 새로 시작되는 것처럼 보일 뿐이며 h5 파일안에는 571번의 업데이트에 대한 파라미터가 저장되어 있을 것으로 예상됩니다.

만약 fit_generator 함수가 572번째 epoch인 것을 인지하도록 학습하려면 fit_generator 함수 호출시에 initial_epoch=572 를 인자값으로 추가하시면 됩니다.

좀더 자세한 인자값에 대한 설명은 아래 문서의 fit_generator 함수 부분을 참조하세요.

좋은 하루 되세요~.

감사합니다.