[DL] PyTorch: nn.ModuleList, nn.ModuleDict, nn.Sequential
nn.ModuleList, nn.ModuleDict, nn.Sequential 사용법 및 유의사항nn.ModuleList, nn.ModuleDict, nn.Sequential는 모델의 block (=submodule) 구성에 매우 유용한 클래스들임.nn.Sequentialnn.Sequential은 여러 모듈을 순차적으로 실행할 수 있도록 하는 PyTorch 클래스임.간단한 네트워크 구조를 만들 때 유용하며, 모듈을 정의한 순서대로 순차적으로 적용함.사용법:import torchimport torch.nn as nnmodel = nn.Sequential( nn.Linear(10, 20), nn.ReLU(), nn.Linear(20, 10))print(model)위 예제에서nn.Sequent..
2024. 5. 17.