jit trace comparisons snuck into torchscript part of validate.py, fixed

2 years ago · 56596e4e84
parent 8ad4bdfa06
commit 56596e4e84
1 changed files with 1 additions and 1 deletions
--- a/validate.py
+++ b/validate.py
@ -181,7 +181,7 @@ def validate(args):

    if args.torchscript:
        torch.jit.optimized_execution(True)
-        model = torch.jit.trace(model, example_inputs=torch.randn((args.batch_size,) + data_config['input_size']))
+        model = torch.jit.script(model)
    if args.aot_autograd:
        assert has_functorch, "functorch is needed for --aot-autograd"
        model = memory_efficient_fusion(model)