[feature] Add auxiliary classification - Githubissues

boostcampaitech3 / level2-semantic-segmentation-level2-cv-17

[2022.04.25 ~ 2022.05.12] Recycle Trash Semantic Segmentation Competition - 부스트캠프 AI Tech 3기

4 stars 2 forks source link

[feature] Add auxiliary classification #39

Closed omocomo closed 2 years ago

omocomo commented 2 years ago

What is this PR?

smp에서 aux_params를 이용해 auxiliary classification을 함께 학습시키는 코드를 작성했습니다.

Changes

config.yaml : aux_params, cls_criterion 추가
dataset.py : getitem에서 labels도 함께 return
loss.py : cls_criterion의 loss 지정
train.py : aux_params의 None 여부에 따라 loss control, wandb에 추가된 loss logging

To reviewers

[ ] config.yaml의 aux_params 작성 시 tab으로 들여쓰기 합니다. aux_params가 None이 아닐 때만 학습 시 적용되며, aux_params 사용 시에는 cls_criterion도 꼭 지정해주어야 합니다.
[ ] 우선 코드 작성은 했으나, 효과를 보기 위한 시각화와 적절한 사용을 위한 실험이 더 필요합니다.