colab에서 dreambooth 학습

  프로젝트를 진행하고 있었지만 그동안 image 분류 작업에 상당히 기간이 오래 걸려서 딱히 진행된 부분이 없었는데 오늘 교수님이 골라주신 이미지를 토대로 학습에 들어갔다.  그동안 colab용 dreambooth를 로컬에서 돌리겠다고 jupyter notebook 으로 계속 시도하고 삽질만 하다가 다시 colab으로 돌아오니 너무 쉽게 다 작동 되었다. 그래도 의문점이 들었던 부분은 concept_lists.json 파일에 모든 label에 대한 정보를 집어넣고 accelarate로 실행시키면 한번에 학습이 되지 않을까 였는데 terminal 에서 직접 실행하도록 만들고 변수를 json에서 읽어오는게 아니라 직접 정해주니 json을 만들지 않고도 학습을 진행 할 수 있었다. 

 

  오늘 한 과정은 학습에서 모르는 부분이 아니라 최종적으로 학습 모델을 Script를 짰다. 기존에 모델에 대해 계속해서 봤던 경험이 도움이 된것 같았다. 중간중간 error를 살펴봐야 진행이 되는 것도 다시 느꼈다. 논문작성까지 한다고 하는데 잘 마무리가 되었으면 좋겠다.