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.