본문 바로가기

Tech/파이썬

Pytorch 프로파일러

Lyken17/pytorch-OpCounter

 

GitHub - Lyken17/pytorch-OpCounter: Count the MACs / FLOPs of your PyTorch model.

Count the MACs / FLOPs of your PyTorch model. Contribute to Lyken17/pytorch-OpCounter development by creating an account on GitHub.

github.com

 

 

[NVIDIA PyProf] FLOPs 측정하기

 

[NVIDIA PyProf] FLOPs 측정하기

PyProf란? NVIDIA에서 만든 profile 툴. PyTorch 모델과 GPU 성능을 profile한다. 1. GitHub로 설치하기 클론합니다 git clone https://github.com/NVIDIA/PyProf.git PyProf로 들어간 후 PyProf를 설치합니다 pip install . 잘 설치

mulkkog.tistory.com

 

 

 

Pytorch Resource & 모델 구조 Profiler 도구 (torch profiler)

 

Pytorch Resource & 모델 구조 Profiler 도구 (torch profiler)

Introduction 딥러닝 학습을 잘(?)한다는 것을 정의하기는 어렵지만, 더 빠른 시간 안에 많은 양을 학습하는 것은 매우 중요하다. 딥러닝의 모델은 다수의 layer로 구성되어 있기 때문에, 각 layer의 결

devhwi.tistory.com

 

 

PYTORCH 프로파일러(PROFILER)

 

PyTorch 프로파일러(Profiler)

이 레시피에서는 어떻게 PyTorch 프로파일러를 사용하는지, 그리고 모델의 연산자들이 소비하는 메모리와 시간을 측정하는 방법을 살펴보겠습니다. 개요: PyTorch는 사용자가 모델 내의 연산 비용

tutorials.pytorch.kr

 

 

'Tech > 파이썬' 카테고리의 다른 글

Bias / torch.empty  (0) 2024.01.22
Python Random Number Generator/난수 생성  (0) 2024.01.19
Pytorch 실험 계획  (0) 2023.12.15
AI 경진대회  (0) 2023.11.13









>