728x90 반응형 module2 [PyTorch] Custom Model 과 torch.nn.Module의 메서드들. Custom Model 만들기 1. top-level attribute로 sub-module추가. 가장 간단하게 sub-module (layer or activation)을 추가하는 방법임. 주의할 것은 top-level attribute가 아닌, list나 dictionary 인스턴스 로 싸고 있는 형태로 추가될 경우, parameters를 통해 제대로 찾지 못하며, 이는 optim 등에서 해당 파라메터나 sub-module의 파라메터를 추적하지 못하는 문제점이나 GPU로 이동시킬 때 해당 파라메터를 제대로 이동시키지 못하는 문제를 일으키게 된다.이에 대한 부분은 다음 URL을 참고할 것.2024.05.17 - [분류 전체보기] - [DL] PyTorch: nn.ModuleList, nn.ModuleD.. 2024. 4. 12. [Python] Module, Package and Library (+ Framework) 필요한 개념 및 용어 정리Library and Frameworkhttps://dsaint31.tistory.com/452 [Programming] Library vs. Framework이 둘을 명확히 구분하는 정의는 사실 없다고 봐야한다. 개인적으로 호출 시점과 같은 흐름에 대한 자유도로 나누는 견해가 가장 맞는 구분 같지만... 우선 두 용어가 가지는 대략적인 느낌의 차dsaint31.tistory.com package.version2024.09.19 - [Python] - [Etc] SW Version: Semantic Versioning + package.version [Etc] SW Version: Semantic Versioning + package.versionSemantic Versionin.. 2024. 2. 3. 이전 1 다음 728x90 반응형