mirrors/pytorch-image-models
1
0
Fork 0
mirror of https://github.com/huggingface/pytorch-image-models.git synced 2025-06-03 15:01:08 +08:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Ross Wightman 2022-07-31 21:13:56 -07:00
parent 8ad4bdfa06
commit 56596e4e84
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
validate.py
View File

@ -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)