[examples] Fix LeNet model parameters issue.

buddy-compiler · Feb 27, 2024 · 74a4901 · 74a4901
1 parent 10d03c7
commit 74a4901
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/BuddyLeNet/buddy-lenet-import.py b/examples/BuddyLeNet/buddy-lenet-import.py
@@ -36,7 +36,7 @@
     )
 
 model = LeNet()
-torch.load(model_path + "/lenet-model.pth")
+model = torch.load(model_path + "/lenet-model.pth")
 model = model.eval()
 
 # Initialize Dynamo Compiler with specific configurations as an importer.