Torchvision ssd not compatible with torch.fx, how can I make it work?


I would like to trace torchvision’s ssdlite320_mobilenet_v3_large however I am getting this issue:

torch.fx.proxy.TraceError: Proxy object cannot be iterated. This can be attempted when used in a for loop or as a *args or **kwargs function argument.

Likely due to some dynamic control flow.
How can I debug this, find the problematic area?
I am not sure this model truly uses dynamic control flow, maybe there is a way to make it work using a custom Tracer?