Yuliang commited on
Commit
f191c89
1 Parent(s): 69cb8a3

Update apps/infer.py

Browse files
Files changed (1) hide show
  1. apps/infer.py +1 -1
apps/infer.py CHANGED
@@ -128,7 +128,7 @@ def generate_model(in_path, model_type):
128
 
129
  optimizer_smpl = torch.optim.Adam(
130
  [optimed_pose, optimed_trans, optimed_betas, optimed_orient],
131
- lr=1e-2,
132
  amsgrad=True,
133
  )
134
  scheduler_smpl = torch.optim.lr_scheduler.ReduceLROnPlateau(
 
128
 
129
  optimizer_smpl = torch.optim.Adam(
130
  [optimed_pose, optimed_trans, optimed_betas, optimed_orient],
131
+ lr=1e-3,
132
  amsgrad=True,
133
  )
134
  scheduler_smpl = torch.optim.lr_scheduler.ReduceLROnPlateau(