multiple inputs

[vdsw]

Thank you so much for sharing the code. My model have two inputs, now, how does thop handle ?

[HaoKang-Timmy]

You could use onnx or jit to profile. Or you may want to try to use this https://github.com/HaoKang-Timmy/torchanalyse