ptflops로 PyTorch 네트워크 분석하기

GitHub - sovrasov/flops-counter.pytorch: Flops counter for convolutional networks in pytorch framework Flops counter for convolutional networks in pytorch framework - GitHub - sovrasov/flops-counter.pytorch: Flops counter for convolutional networks in pytorch framework github.com 딥 러닝 모델을 개발하거나 실험 분석을 할 때 네트워크의 연산량과 패러미터 수를 측정하는 것은 필수입니다. 만약 PyTorch로 구현된 네트워크의 연산량과 패러미터 수를 알고 싶다면 ptflops라는 라이브러리..

2022.03.18