turn off autograd on model parameters

app.py

@@ -11,8 +11,10 @@ with open("imagenet_classes.txt", "r") as f:
 
 
 # load a resnet18 model pretrained on ImageNet
-model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet18', pretrained=True)
-model.eval()
+# and turn off autograd on model's parameters
+model = torch.hub.load('pytorch/vision:v0.10.0', 'resnet18', pretrained=True).eval()
+for param in model.parameters():
+    param.requires_grad = False
 
 
 # preprocess data