multiple inputs
vdsw opened this issue · comments
Thank you so much for sharing the code.
My model have two inputs, now, how does thop handle ?
You could use onnx or jit to profile. Or you may want to try to use this https://github.com/HaoKang-Timmy/torchanalyse