Spaces:

Yuliang
/

ICON

Runtime error

Yuliang commited on Sep 10, 2022

Commit

f191c89

•

1 Parent(s): 69cb8a3

Update apps/infer.py

Files changed (1) hide show

apps/infer.py CHANGED Viewed

@@ -128,7 +128,7 @@ def generate_model(in_path, model_type):
         optimizer_smpl = torch.optim.Adam(
             [optimed_pose, optimed_trans, optimed_betas, optimed_orient],
-            lr=1e-2,
             amsgrad=True,
         )
         scheduler_smpl = torch.optim.lr_scheduler.ReduceLROnPlateau(

         optimizer_smpl = torch.optim.Adam(
             [optimed_pose, optimed_trans, optimed_betas, optimed_orient],
+            lr=1e-3,
             amsgrad=True,
         )
         scheduler_smpl = torch.optim.lr_scheduler.ReduceLROnPlateau(