Pytorch/Tutorials
Tutorial[7] - Train & Validation & Matric
딥러닝의 이론적인 내용은 최대한 배제하고 Pytorch를 활용하여 코딩할 수 있도록 필수로 알아야하는 내용에 대해서만 초점을 두었습니다. Pytorch 1.4 버전을 기준으로 공식 홈페이지에 나와있는 튜토리얼을 참고하였으며, 데이터 로드, CNN모델 설계, 학습, 추론까지의 일련의 과정을 경험해보는것이 Tutorial의 목표입니다. Train 이전에 학습했던 모든 단계가 선행되어야만 학습을 진행할 수 있다. 여기는 의외로 코딩하기가 쉬운데, 앞서 배웠던것들을 하나씩 나열하기만 하면 된다. 프로세스 : # data_loader는 CustomDataSet에서 call한것을 사용. model.train() # 학습시 반드시 모드 변경을 해줄것. for EPOCH in range(1, EPOCHS): epoch..
2020. 10. 4. 00:50
최근댓글