I have been trying to create an fx graph from a module using torch.fx.symbolic_trace(). This often results in many errors because of the symbolic tracing with Proxy objects. In many cases, it would work much better to trace with real tensors, to “freeze” dynamic logic parts of the code and avoid type errors. This would be similar to how torch.jit.trace works.
Is there some way to do this? Maybe a different implementation of Proxy?