[Feature Request] Add model summary for Parameters, GFLOPs and FPS numbers

tomas-gajarsky / facetorch

Python library for analysing faces using PyTorch

Apache License 2.0

503 stars 45 forks source link

[Feature Request] Add model summary for Parameters, GFLOPs and FPS numbers #29

Closed hellojialee closed 1 year ago

tomas-gajarsky commented 1 year ago

I added the number of parameters of each model to the model tables in README v0.1.5.

Computing the FLOPS needs to be done on the original model files, before exporting them to TorchScript, therefore I will have to see if I can still go easily back and get that info.

hellojialee commented 1 year ago

@tomas-gajarsky thumb up!