From e72a7a52a5dbd1a62d062fe6e01f3addfa80fb69 Mon Sep 17 00:00:00 2001
From: William Wen <williamwen@fb.com>
Date: Wed, 26 Apr 2023 00:35:21 +0000
Subject: [PATCH] use better model for torch.compile tutorial

---
 intermediate_source/torch_compile_tutorial.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/intermediate_source/torch_compile_tutorial.py b/intermediate_source/torch_compile_tutorial.py
index fcea4ed6611..2f5d2e966e1 100644
--- a/intermediate_source/torch_compile_tutorial.py
+++ b/intermediate_source/torch_compile_tutorial.py
@@ -117,9 +117,9 @@ def generate_data(b):
 
 N_ITERS = 10
 
-from torchvision.models import resnet18
+from torchvision.models import densenet121
 def init_model():
-    return resnet18().to(torch.float32).cuda()
+    return densenet121().to(torch.float32).cuda()
 
 ######################################################################
 # First, let's compare inference.