From 19f1ea2ecda58ab73286d9a6ac837c683a3e7b31 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 16 Jun 2025 13:07:25 +0200
Subject: [PATCH] fix : frozen param

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 3617217..860ef93 100644
--- a/main.py
+++ b/main.py
@@ -216,7 +216,7 @@ def run_duo(args):
     val_loss=[]
     #init training
     loss_function = nn.CrossEntropyLoss()
-    optimizer = optim.SGD(model.parameters(), lr=0.001, momentum=0.9)
+    optimizer = optim.Adam(model.parameters(), lr=0.001, momentum=0.9)
     #train model
     for e in range(args.epoches):
         loss, acc = train_duo(model,data_train,optimizer,loss_function,e)
-- 
GitLab