Shakerato

pytorch code bottleneck analysis 본문

Research

pytorch code bottleneck analysis

Shakeratto 2021. 5. 26. 14:36

파이토치 개발 후 코드의 어느 부분에서 계산에 많은 시간이 소요되는지 분석하고,

이를 통해 코드 성능을 향상하는데 도움이됨

 

python -m torch.utils.bottleneck /path/to/source/script.py [args]

 

Reference: https://pytorch.org/docs/stable/bottleneck.html

 

torch.utils.bottleneck — PyTorch 1.8.1 documentation

torch.utils.bottleneck torch.utils.bottleneck is a tool that can be used as an initial step for debugging bottlenecks in your program. It summarizes runs of your script with the Python profiler and PyTorch’s autograd profiler. Run it on the command line

pytorch.org

 

저작자표시 비영리

'Research' 카테고리의 다른 글

python 2 to python3 automatic change tool (2to3)  (0) 2021.02.08
HTK, SOX 설치 (ubuntu 18.04 64bit)  (0) 2021.02.08
Check valid webcams with cv2, python  (0) 2020.11.18
Anaconda on Linux, root can't activate my env  (0) 2020.10.18
initialize the empty list in the dictionary' in Python  (0) 2020.09.05
'Research' Related Articles more
Comments