From 56596e4e843f4b3a2a60fc736d17a01e90b7efef Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 31 Jul 2022 21:13:56 -0700 Subject: [PATCH] jit trace comparisons snuck into torchscript part of validate.py, fixed --- validate.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/validate.py b/validate.py index fd55d408..a4d41868 100755 --- 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)