Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[FEAT] 모델 학습 코드 pytorch-template 포팅 #81

Closed
2 tasks done
2018007956 opened this issue Mar 8, 2024 · 2 comments
Closed
2 tasks done

[FEAT] 모델 학습 코드 pytorch-template 포팅 #81

2018007956 opened this issue Mar 8, 2024 · 2 comments
Assignees
Labels
enhancement New feature or request model

Comments

@2018007956
Copy link
Contributor

2018007956 commented Mar 8, 2024

Background

  • 더 좋은 코드에 대한 고민
  • 모듈화가 잘 되어있는 pytorch-template을 가져와서 사용하고자 함
  • 관리 용이성, 체계적이고 생산적인 모델 학습 환경 구축, 하드 코딩 시간 단축을 위함
  • 고민이 되는 부분
    • MLflow에서 자동으로 세세한 로깅을 해주는 Autolog는 pytorch.nn.Module을 지원하지 않음
    • Pytorch Lightning은 지원함
    • 이 기능을 사용하기 위해 pytorch-lightning template을 사용해 보는 건 어떨까?

To do

  • pytorch-lightning template에 대한 고민
  • 선정된 template을 우리의 목적에 맞게 수정하여 사용
@2018007956 2018007956 added enhancement New feature or request model labels Mar 8, 2024
@2018007956
Copy link
Contributor Author

모델 학습 보다 서비스 배포 단계까지의 구현(프론트 및 백엔드 개발)에 더욱 힘을 쏟고 있는 상황이기에
시간 상 pytorch-lightning 변환 작업은 진행하지 않기로 하였습니다.
모델 학습 시 로깅이 필요한 값들을 수동으로 지정해 주어야 합니다.

@2018007956
Copy link
Contributor Author

같은 작업을 하는 이슈가 #86 에 생성되었습니다.
PR #87 #89 참고부탁드립니다.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
enhancement New feature or request model
Projects
None yet
Development

No branches or pull requests

1 participant