딥러닝 flops 계산 코드

    [모델 계산량] FLOPs 계산하는 방법 (How to calculate model FLOPs?)

    임베디드 환경에 딥러닝 모델을 올리는 것을 " 포팅(Porting) " 이라고 한다. 포팅할 때 가장 중요한 것은, 해당 모델 FLOPs를 고려해야한다는 것이다. 예를 들어 Ambarella사의 CV22에 딥러닝 모델을 포팅하기 위해서는 최대 3~4 GFLOPs가 허용치이다. FLOPs가 포팅에 절대적인 기준은 아니지만, 엄청 중요한 기준이 된다는 것은 확실하다. 그러므로 우리는 FLOPs를 자체적으로 계산할 줄 알아야한다. 그래서!!! 아래 내용을 한 문장으로 먼저 요약하자면... pthflops를 사용 하면 된다! " pthflops " install 하기 https://github.com/1adrianb/pytorch-estimate-flops GitHub - 1adrianb/pytorch-esti..