본문 바로가기

부스트캠프 AI Tech/Pytorch

Pytorch - basic

Pytorch

  • Define by Run - runtime 중 graph 생성 
  • GPU support, good API, easy debugging
  • 기본적으로 numpy의 사용법과 매우 유사

cpu와 gpu중 무엇을 사용할지 선택
view, reshape

  • view, reshape은 contiguity(연속적인 데이터의 저장) 보장의 차이

view, memory주소 보장
reshape, memory 보장 x
unsqueeze, parameter의 위치에 1 dim 추가
squeeze, 크기가 1인 dim 삭제
mm, 행렬곱 

 

'부스트캠프 AI Tech > Pytorch' 카테고리의 다른 글

Trouble shooting  (0) 2022.01.27
Hyperparameter Tuning  (0) 2022.01.27
Model Control  (0) 2022.01.26
Pytorch - Project template  (0) 2022.01.24