Deep Learning/PyTorch

PyTorch 란?

green_ne 2022. 1. 28. 11:37

# PyTorch

딥러닝 모델을 짜기 위한 프레임워크.

예전에는 Julia, Keras 등등 수많은 딥러닝 프레임워크가 존재했지만, PyTorch와 TensorFlow가 남았다고 볼 수 있다.

여기서 Keras는 어떤 기능이 있기 보다는 Wrapper로서, 고수준 API라고 볼 수 있다.

 

딥러닝 프레임워크가 필요한 이유는 기초부터 다짜서, 프로젝트를 완성하기에는 너무 오랜시간이 걸리기 때문이다.

하지만 실제로 기본 바닥부터 짜기도 한다. 만약 원한다면, <밑바닥부터 시작하는 딥러닝 3>를 보는 것을 추천한다.

 

TensorFlow PyTorch
Define and run Define by Run
Dynamic Computation Graph (DCG)
static graph 그려서, 나중에 실행시점에 역전파가 일어남.
graph를 작성하는 코드를 작성한 후, 실행시점에 data feed.
production & cloud & multi-GPU 지원.
실행시점에 graph를 그려줌.
개발 과정/ 아이디어 구현/ Pythonic code 용이.
NumPy + AutoGrad + Function

 

# PyTorch 공식 문서

https://pytorch.org/docs/stable/index.html
 

PyTorch documentation — PyTorch 1.10.1 documentation

Shortcuts

pytorch.org

 

반응형