We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 35125b1 commit 8c0785eCopy full SHA for 8c0785e
intermediate_source/torch_compile_tutorial.py
@@ -117,9 +117,9 @@ def generate_data(b):
117
118
N_ITERS = 10
119
120
-from torchvision.models import resnet18
+from torchvision.models import densenet121
121
def init_model():
122
- return resnet18().to(torch.float32).cuda()
+ return densenet121().to(torch.float32).cuda()
123
124
######################################################################
125
# First, let's compare inference.
0 commit comments