I have a similar kind of problem with my model. To deal with variable size input I’ve implemented a custom collate function. Similar to this 55
I want to parallelize my model too.
Did this work for you?? It would be helpful if you can explain in a bit detail. Thankyou so much in advance.