fix model

beeb89gang · Apr 14, 2024 · a34d5a3 · a34d5a3
1 parent 966784f
commit a34d5a3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mACHINE-LEARNINGS/train.py b/mACHINE-LEARNINGS/train.py
@@ -61,8 +61,10 @@ 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=.1, alpha=.99, eps=1e-08, weight_decay=0, momentum=0, centered=False)
-    loss_function = torch.nn.HingeEmbeddingLoss()
+    # 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)
+    # loss_function = torch.nn.HingeEmbeddingLoss()
+    loss_function = torch.nn.BCELoss()
 
     """
     TRAINING PHASE