Graph

이전 글: [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..