Open muzafferkal opened 1 year ago
Hey there @muzafferkal :wave:
My apologies about the very late reply! Could you share a minimal reproducible snippet? I understand that your issue is with a model that has mutiple inputs to custom PyTorch modules. But additional context & snippet would help to come up with the best solution!
Cheers :v:
Bug description
If a network returns multiple outputs, the forward hook at crawler.py:181 crashes because the out parameter is a tuple and not a tensor so it doesn't support .size() method. Are module returning multiple outputs supported and I'm doing wrong or is this a bug? Thanks.
Code snippet to reproduce the bug
Error traceback
Environment