fix : frozen param

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)