just moving computer i actually have no idea what i'm committing lmao

beeb89gang · Apr 20, 2024 · 38b988f · 38b988f
1 parent f087727
commit 38b988f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/mACHINE-LEARNINGS/eval.py b/mACHINE-LEARNINGS/eval.py
@@ -38,15 +38,14 @@ def eval(image):
         - loss function: binary cross entropy loss
     """
     model = DankCNN()
-    model.load_state_dict(torch.load('models/trained/model.pt'))
+    model.load_state_dict(torch.load('models/trained/model.pt', map_location=torch.device('cpu')))
     model = model.to(device)
 
     model.eval()
     with torch.no_grad():
         image = image.unsqueeze(0)
         output = model(image.to(device))
         prediction = output.data
-        print(prediction.item())
         print('This meme is', cls[0 if prediction.item() < .5 else 1])
 
 

diff --git a/mACHINE-LEARNINGS/train.py b/mACHINE-LEARNINGS/train.py
@@ -62,7 +62,7 @@ def execute(train_set_size, batch_size, lr, epochs, is_verbose, weight_decay):
     model = DankCNN()
     model = model.to(device)
     # optimizer = torch.optim.RMSprop(model.parameters(), lr=.001, alpha=.99, eps=1e-08, weight_decay=0, momentum=0, centered=False)
-    optimizer = torch.optim.Adam(model.parameters(), lr=0.001, betas=(0.9, 0.999), eps=1e-08, weight_decay=0)
+    optimizer = torch.optim.Adam(model.parameters(), lr=0.0001, betas=(0.9, 0.999), eps=1e-08, weight_decay=0)
     # loss_function = torch.nn.HingeEmbeddingLoss()
     loss_function = torch.nn.BCELoss()