Lyken17/pytorch-memonger

support dynamic graph with torch.compile() api

lxysl opened this issue · 0 comments

lxysl commented

Now we have torch.compile() api in PyTorch 2.0+, is it able to use any PyTorch model inherited from nn.Module() instead of just nn.Sequential()?