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