vhad/body-pose-animation
1
0
Fork 0
mirror of https://github.com/gosticks/body-pose-animation.git synced 2025-10-16 11:45:42 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

play with config

Browse Source
This commit is contained in:
Wlad 2021-02-05 22:18:53 +01:00
parent 2a9f7f1676
commit d230c3320c
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
config.yaml
View File

@ -28,10 +28,10 @@ pose:
weight: 0.001 weight: 0.001
bodyPrior: bodyPrior:
enabled: true enabled: true
weight: 0.5 weight: 2
anglePrior: anglePrior:
enabled: true enabled: true
weight: 0.01 weight: 0.001
# idxWeights: # idxWeights:
# - 0.5 # - 0.5
# - 0.5 # - 0.5

2
train_pose.py
View File

@ -59,7 +59,7 @@ def train_pose(
vposer = VPoserModel() vposer = VPoserModel()
vposer_layer = vposer.model vposer_layer = vposer.model
vposer_params = vposer.get_vposer_latent() vposer_params = vposer.get_vposer_latent()
parameters.append(vposer_params) # parameters.append(vposer_params)
if useAnglePrior: if useAnglePrior:
angle_prior_layer = AnglePriorsLoss(dtype=dtype, device=device) angle_prior_layer = AnglePriorsLoss(dtype=dtype, device=device)