diff --git a/main.py b/main.py
index 860ef937f1917508cc1933e24b71d36bdb48b23c..8cdbb714437edbc344d1dd1510401c95ef5b184b 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.Adam(model.parameters(), lr=0.001, momentum=0.9)
+    optimizer = optim.Adam(model.parameters(), lr=0.001)
     #train model
     for e in range(args.epoches):
         loss, acc = train_duo(model,data_train,optimizer,loss_function,e)