Spaces:
Sleeping
Sleeping
| wandb_project: 'EMAGE++' | |
| exp_name: 'debug' | |
| debug: False | |
| wandb_entity: '' | |
| wandb_key: "" | |
| # wandb_log_dir: '/content/outputs/wandb' | |
| output_dir: "../outputs/" | |
| log_period: 1 | |
| seed: 42 | |
| resume_from_checkpoint: null | |
| test: False | |
| data: | |
| name_pyfile: "datasets.beat2" | |
| class_name: "BEAT2Dataset" | |
| train_bs: 64 | |
| meta_paths: | |
| - "./datasets/data_json/beat2_s20_l128_speaker2.json" | |
| test_meta_paths: | |
| - "./datasets/data_json/beat2_s20_l128_speaker2.json" | |
| pose_norm: False | |
| pose_length: 128 | |
| stride: 20 | |
| test_length: 128 | |
| model: | |
| name_pyfile: "models.camn_audio.modeling_camn_audio" | |
| class_name: "CamnAudioModel" | |
| pose_fps: 15 | |
| motion_f: 256 | |
| pose_dims: 258 | |
| pose_rep: "smplx" | |
| body_dims: 78 | |
| hands_dims: 180 | |
| audio_rep: wave16k | |
| audio_sr: 16000 | |
| audio_fps: 16000 | |
| audio_norm: False | |
| audio_f: 128 | |
| speaker_f: 16 | |
| speaker_dims: 1 | |
| hidden_size: 512 | |
| n_layer: 4 | |
| dropout_prob: 0.1 | |
| seed_frames: 4 | |
| joint_mask: "local_upper" | |
| validation: | |
| validation_steps: 500 | |
| test_steps: 500 | |
| visualization: False | |
| evaluation: False | |
| wandb: False | |
| solver: | |
| gradient_accumulation_steps: 1 | |
| gradient_checkpointing: False | |
| max_train_steps: 100000 | |
| max_grad_norm: 0.0 | |
| # lr | |
| learning_rate: 3e-4 | |
| scale_lr: False | |
| lr_warmup_steps: 0 | |
| lr_scheduler: 'constant' | |
| # optimizer | |
| use_8bit_adam: False | |
| adam_beta1: 0.9 | |
| adam_beta2: 0.999 | |
| adam_weight_decay: 0.0 | |
| adam_epsilon: 1.0e-8 |