Skip to content

yongchoooon/YCHPytorchTemplate

Repository files navigation

YCHPytorchTemplate

딥러닝 모델에 데이터셋을 학습시키는 Customized Pytorch template

Contents

  • dataset handling, data loader, model, trainer, logger 등 DL 학습을 위한 코드 구조 이해
  • 원하는 dataset을 불러와서 학습에 활용
  • 나만의 data loader, model 코드를 작성하기
  • 학습을 진행하고 logging, tensorboard writing 등 기록 및 저장 과정 익히기
  • 학습 과정 중 slack으로 log 알림 보내는 기능 추가

Datasets

Assignment

Dog Breed Classification

  • Notion: Dog Breed Classification
    • DensetNet 코드 구현 및 학습
    • VGG 모델 학습 이후 Generation 기법 적용해 성능 올리기 연습
    • Transfer Learning으로 성능 올리기

About

PyTorch Template Customized by Yongchoooon

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages