-
Notifications
You must be signed in to change notification settings - Fork 5
(4.27) backbone 실험
HyeonJin edited this page Apr 28, 2021
·
1 revision
모델 | 배치 | 시드 | time/step | epoch | loss | val_loss | val_mIoU | LB score |
---|---|---|---|---|---|---|---|---|
DeepLabV3+, resnext50 | 16 | 77 | 4.8s | 10 | 0.108 | 0.283 | 0.463 | 0.5754 |
DeepLabV3+, seresnext50 | 16 | 77 | 6.8s | 16 | 0.057 | 0.280 | 0.476 | 0.5670 |
DeepLabV3+, resnext50 | 8 | 77 | 2.4s | 9 | 0.098 | 0.294 | 0.451 | 0.5795 |
- scheduler: CosineAnnealingWarmup(LR_start=2e-6, LR_max=1e-4, T_0=20, T_up=2)
- batch_size를 8로 주면 마지막(327번째) 배치에서 size관련 에러 발생
ㄴ train_loader에서 drop_last=True로 해결했으나 원인파악을 못함
- V3에 비해 V3+가 파라미터 수도 작고, 러닝타임도 빠름
- smp 모델에서 인코더 부분만 timm 모델로 교체하려 했으나, 동일 모델이라도 weight값이 아예 달랐음 (일단 시도 예정)
- PSPNet : eff-b0(3.2s), eff-b3(3.5s), resnet34(6.0s)
- DeepLabV3+ : resnext50(4.8s), eff-b0(6.7s), seresnext50(6.8s), eff-b3(9.5s), effi-b5(15.5s)