hunkim / PyTorchZeroToAll

Simple PyTorch Tutorials Zero to ALL!
http://bit.ly/PyTorchZeroAll
3.89k stars 1.2k forks source link

Propose some modified codes for PyTorchZeroToAll #27

Open ghlee0304 opened 6 years ago

ghlee0304 commented 6 years ago

몇 가지 경고 문구를 수정한 것 ".data[0] -> .item()" cpu version에서 실행이 안되는 것 "num_workers=0 -> num_workers=0" dimension문제 수정 label = label.unsqueeze(0) parameter로 실행이 안되는 문제 12-4 등의 수정을 제안드립니다.