/dl_baseline

pytorch 기반 dl baseline을 구성합니다

Primary LanguagePython

DL_classification_baseline

pytorch 기반 DL(DNN) classification baseline입니다.(Example : MMIST)

각 요소 역할

  • model.py : model이 정의된 파일
  • trainer.py : model 학습하는 파일
  • train.py : hyperparameter를 입력받고, model, trainer, loader 선언 이후 학습
  • predict.py : model과 input을 입력받아 predict 수행
  • utils.py : 예시 코드 구현을 위해 MNIST 실습을 위한 load, preprocessing 등의 과정 수행
  • loader.py : torch.Dataset과 torch.Dataloader를 이용하여 data 적재