tools for FLOPs and parameters calculation of pytorch
from torchvision.models import resnet18
from coreOperator import get_params, get_flops
model = resnet18()
print(get_params(model,input_size=(1,3,64,64),types="G"), get_flops(model, types="G"))
-
Notifications
You must be signed in to change notification settings - Fork 0
tirtile/torchFP
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
tools for FLOPs and parameters calculation of pytorch
Resources
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published