Input of the model must be one?

sovrasov / flops-counter.pytorch

Flops counter for convolutional networks in pytorch framework

MIT License

2.82k stars 307 forks source link

Input of the model must be one? #93

Closed Y1YU closed 2 years ago

Y1YU commented 2 years ago

It seems that ptflops only work when the input of the model only be one. If my model has two inputs, one is (3, 224, 224), one is (17), then it will report an error.

sovrasov commented 2 years ago

Hi! See https://github.com/sovrasov/flops-counter.pytorch/issues/14

Y1YU commented 2 years ago

Thank you for your answer. And I want to ask if 1 GFLOPs == 2 GMacs？

sovrasov commented 2 years ago

This was discussed in #16