2024.01.15 DAY78
딥러닝 프로젝트 발표일!!
주말 동안 centerpoint 모델 학습을 진행하였다. 학습 epoch을 조절하고, pretrained weight를 사용해보는 등 여러 방법으로 학습을 진행하였다. 코드를 직접 구현한 것이 아니라 코드를 파악하기 어려웠다. epoch은 어디서 조절해야하는지, class를 바꾸려면 어디를 수정해야하는지 찾는데 시간이 오래걸렸다. 그리고 환경 변수 에러, gpu, docker 관련 에러 등 평소에 자주 보지 못했던 에러들이 많이 발생해서 이를 해결하는 과정도 오래걸렸다.
또한 AWS를 다같이 사용해서 모델을 학습시키다보니 용량이 부족한 문제가 자주 발생했다. 결국 docker container를 하나만 사용하는 방법으로 용량 문제를 해결했다. 사실 완전히 해결한 것은 아니지만 일단 다른 container를 제거해서 용량을 확보했다.
🧬 딥러닝 프로젝트 결과
같은 dataset으로 epoch를 조절하여 학습을 진행해보았다. 또한 mini dataset은 데이터 수가 부족하여 학습이 잘 되지 않는 것 같다고 판단하여 train dataset을 사용하여 학습을 진행해보았다.
그 결과 v1.0-trainval-1의 경우 데이터의 수가 훨씬 많기 때문에 mini의 경우보다 성능이 좋았다.
