[딥러닝] 딥러닝 모델을 간단하게 시각화하는 프로그램, Netron | Pytorch, ONNX
·
Python/Deep Learning
오늘은 자신이 직접 디자인한 딥러닝 모델을 시각화하는 작업을 해보겠습니다. 우선, Pytorch로 구현된 모델을 ONNX로 저장해야 합니다. Pytorch로 ONNX로 저장하기 위해서 아래 명령어를 통해 설치합니다. pip install onnx-pytorch 설치를 완료 후, 자신이 구현한 모델의 객체가 있다면, 아래 코드를 통해 모델을 저장합니다. import torch.onnx input_names = ['Time series data'] output_names = ['Output'] x = torch.zeros(1, 10, 6).to(device) torch.onnx.export(model, x, 'regression_mode.onnx', input_names=input_names, output_..
18진수
'이미지' 태그의 글 목록