Pytorch/Tutorials
Tutorial[5] - model
딥러닝의 이론적인 내용은 최대한 배제하고 Pytorch를 활용하여 코딩할 수 있도록 필수로 알아야하는 내용에 대해서만 초점을 두었습니다. Pytorch 1.4 버전을 기준으로 공식 홈페이지에 나와있는 튜토리얼을 참고하였으며, 데이터 로드, CNN모델 설계, 학습, 추론까지의 일련의 과정을 경험해보는것이 Tutorial의 목표입니다. model Pytorch에서 모델을 생성하는 방법은 Tensorflow보다 쉽다. 그 방법이 매우 직관적이기 때문이다. 모든 함수는 torch.nn이라는 곳에서 사용하며, 모델과 관련된 모든 함수가 이곳에 있다. Create Network : class MyNetwork(nn.Module): def __init__(self, num_class=10): super(MyNetwo..
2020. 10. 2. 15:59
최근댓글