split config for lbfgs and adam

This commit is contained in:
Wlad 2021-02-09 10:48:21 +01:00
parent f0205bf56b
commit 66d5360fe1
2 changed files with 70 additions and 7 deletions

View File

@ -0,0 +1,59 @@
output:
rootDir: "./tests/09-02-21"
prefix: result
smpl:
modelRootDir: ./models
type: smplx # possible options smplx and smpl
gender: MALE # possible options MALE, FEMALE, NEUTRAL
ext: npz
useVposerInit: false
data:
renameFiles: false
rootDir: ./output
personId: 0
sampleImageFormat: "input_%%%%%%%%%%%%_rendered.png"
sampleNameFormat: "input_%%%%%%%%%%%%_keypoints.json"
sampleCoords: !!python/tuple [1080, 1080]
camera:
lr: 0.001
patience: 10
optimizer: Adam
pose:
# device: cuda
lr: 0.05
optimizer: LBFGS # currently supported Adam, LBFGS
iterations: 15
useCameraIntrinsics: true
bodyMeanLoss:
enabled: false
weight: 0.1
bodyPrior:
enabled: true
weight: 4.0
anglePrior:
enabled: true
weight: 0.0001
angleSumLoss:
enabled: true
weight: 0.2
confWeights:
enabled: false
vposerPath: "./vposer_v1_0"
temporal:
enabled: true
iterations: 4
lr: 0.02
preview:
enable: true,
keypoins:
enable: true,
radius: 0.01
color: 1.0, 0.0, 1.0, 1.0
keypoint_torso:
enable: true,
radius: 0.01
color: 1.0, 0.0, 1.0, 1.0
joints:
enable: true
radius: 0.01
color: 0.0, 0.7, 0.0, 1.0

View File

@ -1,6 +1,6 @@
output:
rootDir: "./results/"
prefix: result-
rootDir: "./tests/09-02-21"
prefix: ""
smpl:
modelRootDir: ./models
type: smplx # possible options smplx and smpl
@ -19,10 +19,10 @@ camera:
patience: 10
optimizer: Adam
pose:
device: cuda
lr: 0.5
optimizer: LBFGS # currently supported Adam, LBFGS
iterations: 10
# device: cuda
lr: 0.01
optimizer: Adam # currently supported Adam, LBFGS
iterations: 150
useCameraIntrinsics: true
bodyMeanLoss:
enabled: false
@ -35,10 +35,14 @@ pose:
weight: 0.01
angleSumLoss:
enabled: false
weight: 0.01
weight: 0.02
confWeights:
enabled: false
vposerPath: "./vposer_v1_0"
temporal:
enabled: true
iterations: 30
lr: 0.01
preview:
enable: true,
keypoins: