Update apps/infer.py

apps/infer.py

@@ -126,10 +126,10 @@ def generate_model(in_path, model_type):
             data["global_orient"], device=device, requires_grad=True
         )  # [1,1,3,3]
 
-        optimizer_smpl = torch.optim.SGD(
+        optimizer_smpl = torch.optim.Adam(
             [optimed_pose, optimed_trans, optimed_betas, optimed_orient],
-            lr=1e-3,
-            momentum=0.9,
+            lr=1e-2,
+            amsgrad=True,
         )
         scheduler_smpl = torch.optim.lr_scheduler.ReduceLROnPlateau(
             optimizer_smpl,