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

add weights to prior

Browse Source
This commit is contained in:
Wlad 2021-02-09 14:14:23 +01:00
parent 07542309ef
commit 64d40ebe79
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
train_pose.py
View File

@ -205,6 +205,7 @@ def get_loss_layers(config, device, dtype):
if config['pose']['anglePrior']['enabled']:
extra_loss_layers.append(AnglePriorsLoss(
device=device,
global_weight=config['pose']['anglePrior']['weight'],
dtype=dtype))
if config['pose']['angleSumLoss']['enabled']: