Plot of GMac and Params per layer

sovrasov / flops-counter.pytorch

Flops counter for convolutional networks in pytorch framework

MIT License

2.82k stars 307 forks source link

Plot of GMac and Params per layer #66

Open carlodizon03 opened 3 years ago

carlodizon03 commented 3 years ago

It would be nice to have a dictionary containing each layer/operations' GMac and Params that would be useful to plot and be able to see network bottlenecks.