pytorch
fa92b6a7 - Error when jit.trace/script is used with torch.compile (#91681)

Commit
2 years ago
Error when jit.trace/script is used with torch.compile (#91681) Fixes https://github.com/pytorch/pytorch/issues/93485 ```python import torch from torchvision.models import resnet50 model = resnet50(weights=None) compile_model = torch.compile(model) print(type(compile_model)) example_forward_input = torch.rand(1, 3, 224, 224) c_model_traced = torch.jit.trace(compile_model, example_forward_input) # or torch.jit.script torch.jit.save(c_model_traced, "c_trace_model.pt") ``` Should I raise a warning if a user tries to compile a scripted or traced model as well? It works just fine now on resnet but not sure if it's that something we want to explicitly discourage Pull Request resolved: https://github.com/pytorch/pytorch/pull/91681 Approved by: https://github.com/desertfire
Author
Committer
Parents
Loading