새소식

딥러닝

Training / Inference Design & Model Training Procedure

  • -

일반적인 Training / Inference Design

 

model.py: Architecture가 정의된 클래스

trainer.py: Model을 학습하기 위한 코드

dataloader.py : 데이터를 불러와 전처리를 수행하고 신경망에 넣기 좋은 형태로 변환

train.py: 사용자로부터 hyper-param을 받아 Model과 Trainer, Loader을 선언하고 학습

predict.py: 사용자로부터 Model과 input을 받아 추론을 진행

 

일반적인 Model Training Procedure

 

728x90

'딥러닝' 카테고리의 다른 글

MLE(Maximum Likelihood Estimation)  (0) 2022.06.23
Autoencoder  (0) 2022.06.22
머신러닝 프로젝트 Flow  (0) 2022.05.26
Tensorflow 모델 서비스하기  (0) 2022.05.19
활성화 함수(Activation function)  (0) 2022.05.15
Contents