반응형
EX
-
TENSOR RT _ SIMPLE MNIST 예제뜯고 또 뜯어보는 컴퓨터/TENSORRT 2021. 10. 12. 17:00
안녕하세요. 오늘은 TENSOR RT를 활용한 예시를 적어보려 합니다. TensorRT/samples/python/network_api_pytorch_mnist at master · NVIDIA/TensorRT (github.com) 공식 문서를 참고하였으며, 보기 좋게 Ipynb 를 사용하여 분석하였습니다. IMPORT MODULE import torch import torch.nn as nn import torch.nn.functional as F import torch.optim as optim from torchvision import datasets, transforms from torch.autograd import Variable import numpy as np import os from ..