pytorch

이전 글: [AI/Self-Study] - PyTorch 모델 구조 summary & 그래프화 1 PyTorch 모델 구조 summary & 그래프화 1 TensorFlow에서는 model.summary() 메서드 호출을 통해 모델을 요약해서 layer마다 shape와 같은 정보들을 볼 수 있다. PyTorch에서 model.summary() 처럼 사용할 수 있는 메서드와 모델 구조를 그래프화 하는 방.. lynnshin.tistory.com 저번 글에 이어서 PyTorch 모델 시각화 방법에 대해 알아보자 2. 모델 시각화 확인 2-1) Torchviz !pip install torchviz from torchviz import make_dot import torch x = torch.zeros(1,1..

다음 편 : [AI/Self-Study] - PyTorch 모델 구조 summary & 그래프화 2 PyTorch 모델 구조 summary & 그래프화 2 이전 글: [AI/Self-Study] - PyTorch 모델 구조 summary & 그래프화 1 PyTorch 모델 구조 summary & 그래프화 1 TensorFlow에서는 model.summary() 메서드 호출을 통해 모델을 요약해서 layer마다 shape와 같은 정.. lynnshin.tistory.com TensorFlow에서는 model.summary() 메서드 호출을 통해 모델을 요약해서 layer마다 shape와 같은 정보들을 볼 수 있다. PyTorch에서 model.summary() 처럼 사용할 수 있는 메서드와 모델 구조를 그..

데이터 병렬 처리 (DATA PARALLELISM) tutorials.pytorch.kr/beginner/blitz/data_parallel_tutorial.html device = torch.device("cuda:0") if torch.cuda.device_count() > 1: print("Let's use", torch.cuda.device_count(), "GPUs!") # dim = 0 [30, xxx] -> [10, ...], [10, ...], [10, ...] on 3 GPUs model = torch.nn.DataParallel(model) model.to(device)