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 적재