DAWN / pbnet /blink /opt.yaml
Hanbo-Cheng's picture
update
3894f3b
activation: gelu
archiname: transformerreemb5
audio_latent_dim: 256
batch_size: 82
cuda: true
dataset: hdtf
expname: exps
eye: 'True'
ff_size: 64
first3: 'False'
folder: exps_delta_eye_rope/HDTF_200_onlyeye_fast
gpu: '0'
lambda_kl: 0.001
lambda_rc: 1.0
lambda_rcw: 1.0
lambda_ssim: 1.0
lambdas:
kl: 0.001
rc: 1.0
latent_dim: 256
losses:
- rc
- kl
lr: 0.0001
max_distance: 128
max_len: -1
min_len: -1
modelname: cvae_transformerreemb5_rc_kl
modeltype: cvae
num_buckets: 128
num_epochs: 100000
num_frames: 200
num_layers: 2
num_seq_max: -1
pose_latent_dim: 32
snapshot: 5000