Pytorch 기본

[pytorch] 학습을 위해 train 함수에 꼭 필요한 내용들