[Caffe] Caffe를 이용하여 학습할때 snapshot을 이용하여 이어서 학습하기

[Deep Learning] Caffe를 이용하여 학습할때 snapshot을 이용하여 이어서 학습하기

caffe를 이용하여 학습하다 보면 사용자가 정한 iteration마다 snapshot을 찍게 된다.

만약 내가 10만번 반복학습 했는데 성능이 기대한 만큼 안나오면??

그래서 20만번 까지 돌리면 잘 나오겠다 싶은경우에는??

설마 처음부터 다시 20만번을 돌려야하는가!! 물론 아니다.

caffe 에서는 snapshot을 이용하여 저장된 시점부터 이어서 학습을 가능하게 해 준다.

기존에 사용할 경우에 

caffe.exe --solver=~~~.prototxt

처럼 사용한 경우에 아래와 같이 실할할 때 뒤에 이어서 --snapshot 옵션을 이용해 주면 그 파일을 읽어서 해당 시점부터

다시 이어서 학습을 진행한다.

caffe.exe --solver=~~~.prototxt --snapshot=_iter_100000_~~~.solverstate

단, 이때 solverstate와 함께 caffemodel 파일도 같이 있어야 이어서 진행이 가능하다.