미해결
[개정판] 딥러닝 컴퓨터 비전 완벽 가이드
keras-yolo 모델 학습관련 질문
안녕하세요 선생님
먼저 좋은 강의 정말 감사합니다.
저는 [yolo raccoon custom 강의]를 보고 keras-yolo에 제 데이터에 custom하는 작업을 진행 중인데요.
문제는 제 데이터 양 때문인지 한 1epoch 당 3시간 정도가 걸립니다.
저는 aws 를 사용하고 있는데 이게 오래 학습을 시키면 저절로 서버다운이 되고... 중간에 학습모델 저장이 쉽지가 않습니다.
그래서 2epoch씩 학습시키고 저장하고 -> 저장된 모델을 로드해서 다시 2epoch 학습하고
이런 방식으로 노동을 하고자 하는데 그 방법을 여쭈어보고 싶습니다.
여기서 저장된 모델 이름은 trained_weight_stage_1.h5 입니다. fine-tune은 할 엄두도 못내고 있구요...
1. 2epoch 학습한 trained_weight_stage_1.h5 를 yolo.h5 자리에 넣고 학습시키면 되는 걸까요?
2. 어느정도 학습한 모델을 이어서 그다음 코드인 fine-tune 단계만 진행하려면 어떻게 해야할까요?