PyTorch로 커스텀 Loss 구현
PyTorch를 선택한 이유는 nn.Module 상속 구조가 커스텀 레이어와 손실 함수 구현에 유연했기 때문입니다. 클래스 불균형이 심한 데이터셋에서 Focal Loss를 직접 구현해 적용했고, torch.autograd로 그래디언트를 직접 디버깅하며 수렴 문제를 해결했습니다. 학습 루프는 pytorch-lightning으로 감싸 체크포인트, 로깅, 조기 종료를 표준화했습니다. 동료가 TensorFlow 배경이라 ONNX로 모델을 내보내 함께 추론 파이프라인을 공유했습니다. 이 프로젝트에서 F1이 0.71에서 0.84로 개선됐고, 재현 가능한 학습 환경 구성이 협업 효율에 큰 영향을 준다는 점을 배웠습니다.