PyTorch (1) 썸네일형 리스트형 PyTorch 기초1: 코드짜는 법 PyTorch로 네트워크를 설계하는 방법은 크게 3가지로 나눌 수 있습니다. 1) Barebones PyTorch 2) PyTorch Module API 3) PyTorch Sequential API 각각을 간략히 둘러보겠습니다. 1) Barebones PyTorch # weight initialization conv_w1 = torch.empty((channel_1, C, kernel_size_1, kernel_size_1), dtype=dtype).to(device) conv_b1 = torch.empty(channel_1, dtype=dtype).to(device) conv_w1 = nn.init.kaiming_normal_(conv_w1) conv_b1 = nn.init.zeros_(conv_b1.. 이전 1 다음