Skip to content

Commit 8c0785e

Browse files
williamwen42Svetlana Karslioglu
and
Svetlana Karslioglu
authored
use better model for torch.compile tutorial (#2297)
Co-authored-by: Svetlana Karslioglu <svekars@fb.com>
1 parent 35125b1 commit 8c0785e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

intermediate_source/torch_compile_tutorial.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -117,9 +117,9 @@ def generate_data(b):
117117

118118
N_ITERS = 10
119119

120-
from torchvision.models import resnet18
120+
from torchvision.models import densenet121
121121
def init_model():
122-
return resnet18().to(torch.float32).cuda()
122+
return densenet121().to(torch.float32).cuda()
123123

124124
######################################################################
125125
# First, let's compare inference.

0 commit comments

Comments
 (0)