boostcampaitech4lv23nlp1 / final-project-level3-nlp-03

Multi-Modal Model for DocVQA(Document Visual Question Answering)
3 stars 0 forks source link

refactor: fix metrics & num_workers #3 #17

Closed Ssunbell closed 1 year ago

Ssunbell commented 1 year ago
  1. dataloader에서 처리 속도가 느린 것을 num_workers로 해결함. 4정도면 준수, 8이면 가장 빠르지만 서버 cpu 갯수가 8개이므로 만약에 돌려놓고 다른 작업을 하고 싶다면 4개가 적당해보임
  2. train.py를 돌릴때 trainer를 실행하기전에 metric 함수가 실행되는 것을 trainer 들어가고 나서 실행하게 바꿨음. 그 이유는 train을 실행해보고 train loop가 몇시간 걸릴지 한번 찍어보는 것도 중요하기 때문에 metric은 train loop가 한번 돌고 나서 실행되게 바꿈