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

Update analysis.py

Browse source
This commit is contained in:
Xu Ma 2022-04-19 15:45:58 -04:00
parent 09a60de6bc
commit 5cb88d65e3
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
analysis.py
View file

@ -5,9 +5,10 @@ from fvcore.nn import FlopCountAnalysis
from classification_ScanObjectNN.models import pointMLPElite from classification_ScanObjectNN.models import pointMLPElite
model = pointMLPElite() model = pointMLPElite()
model.eval()
# model = deit_tiny_patch16_224() # model = deit_tiny_patch16_224()
inputs = (torch.randn((1,3,1024)),) inputs = (torch.randn((1,3,1024)))
k = 1024.0 k = 1024.0
flops = FlopCountAnalysis(model, inputs).total() flops = FlopCountAnalysis(model, inputs).total()
print(f"Flops : {flops}") print(f"Flops : {flops}")