Safran/PointMLP
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

Create analysis.py

Browse source
This commit is contained in:
Xu Ma 2022-04-19 11:37:20 -04:00
parent c1d6235405
commit 09a60de6bc
1 changed files with 19 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
analysis.py Normal file
View file

@ -0,0 +1,19 @@
import torch
import fvcore.nn
import fvcore.common
from fvcore.nn import FlopCountAnalysis
from classification_ScanObjectNN.models import pointMLPElite
model = pointMLPElite()
# model = deit_tiny_patch16_224()
inputs = (torch.randn((1,3,1024)),)
k = 1024.0
flops = FlopCountAnalysis(model, inputs).total()
print(f"Flops : {flops}")
flops = flops/(k**3)
print(f"Flops : {flops:.1f}G")
params = fvcore.nn.parameter_count(model)[""]
print(f"Params : {params}")
params = params/(k**2)
print(f"Params : {params:.1f}M")