Machine-Learning-Foundations
diff --git a/‎data/weights.pkl‎
-6.49 MB b/‎data/weights.pkl‎
-6.49 MB
diff --git a/‎data/weights.pth‎
1.49 MB b/‎data/weights.pth‎
1.49 MB
diff --git a/‎src/input_opt.py‎
Lines changed: 2 additions & 2 deletions b/‎src/input_opt.py‎
Lines changed: 2 additions & 2 deletions
@@ -33,7 +33,7 @@ def forward(self, x):
 
 
 if __name__ == "__main__":
-    with open("weights.pth", "rb") as file:
+    with open("./data/weights.pth", "rb") as file:
         weights = torch.load(file)
 
     net = CNN()
@@ -47,4 +47,4 @@ def forward_pass(x):
 
     get_grads = grad(forward_pass)
 
-    # TODO: Optimize an input to maximize that output.
+    # TODO: Optimize an input to maximize that output.