diff --git a/main.py b/main.py
index 3617217e15dded9ae61c5fb8adc7ba3f798e30d8..860ef937f1917508cc1933e24b71d36bdb48b23c 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)