pytorch에서 keras처럼 summary를 확인하는 함수가 있다. 이때 가장 많이 사용하는 함수가 아래의 함수일 것이다. from torchsummary import summary 필자도 torchsummary를 사용하여 model의 summary를 확인했다. 그러나 Hrnet을 설계할 때 값이 리스트 형태로 전달하면서 다음과 같은 오류가 발생했다. File "\torchsummary\torchsummary.py", line 19, in hook summary[m_key]["input_shape"] = list(input[0].size()) AttributeError: 'list' object has no attribute 'size' 그래서 구글에서 검색 중 github에서 해결 방법을 찾았다. ..
☞ 문서의 내용은 가장 하단 참고문헌 및 사이트를 참고하여 필자가 보기 쉽도록 정리한 내용입니다. ☞ 틀린 내용 및 저작권 관련 문의가 있는 경우 문의하시면 수정 및 삭제 조치하겠습니다. 자동 미분 (Autograd : automatic differentiation) autograd package는 Tensor의 모든 연산에 대해 자동 미분을 제공한다. define-by-run framework로, 코드를 어떻게 작성하여 실행하느냐에 따라 역전파가 정의된다는 뜻이며, 역전파는 학습 과정의 매 단계마다 달라진다. torch.Tensor는 package에서 중심이 되는 class이다. .requires_grad 속성을 True로 설정하면, 그 tensor에서 이뤄진 모든 연산들을 추적한다. 계산이 완료된 후..
☞ 문서의 내용은 가장 하단 참고문헌 및 사이트를 참고하여 필자가 보기 쉽도록 정리한 내용입니다. ☞ 틀린 내용 및 저작권 관련 문의가 있는 경우 문의하시면 수정 및 삭제 조치하겠습니다. pytorch가 무엇인가? Pyton기반의 과학 연산 패키지로 아래와 같은 두 집단을 대상으로 만들어졌다. - Numpy를 대체하면서 GPU를 이용한 연산이 필요한 경우 - 최대한의 유연성과 속도를 제공하는 딥러닝 연구 플랫폼이 필요한 경우 Tensors - tensor는 Numpy의 ndarray와 유사하며, GPU를 사용한 연산 가속이 가능하다. 초기화되지 않은 행렬이 생성되면 그 시점에 할당된 메모리에 존재하던 값들이 초기값으로 나타난다. 1. 초기화되지 않은 5x3 행렬을 생성 x = torch.empty(5, ..
내 블로그 - 관리자 홈 전환 |
Q
Q
|
---|---|
새 글 쓰기 |
W
W
|
글 수정 (권한 있는 경우) |
E
E
|
---|---|
댓글 영역으로 이동 |
C
C
|
이 페이지의 URL 복사 |
S
S
|
---|---|
맨 위로 이동 |
T
T
|
티스토리 홈 이동 |
H
H
|
단축키 안내 |
Shift + /
⇧ + /
|
* 단축키는 한글/영문 대소문자로 이용 가능하며, 티스토리 기본 도메인에서만 동작합니다.