| data: | |
| chunk_size: 150 | |
| json_files: | |
| - ./jsons/blow.json | |
| - ./jsons/eyebrow.json | |
| - ./jsons/pout.json | |
| - ./jsons/nersemble_front.json | |
| - ./jsons/yk_cap.json | |
| - ./jsons/xnemo_transfer.json | |
| mount_prefix: /aliyun-oss | |
| num_workers: 4 | |
| oss_prefix: oss://aigcdevwlcb | |
| seed: 42 | |
| val_ratio: 0.1 | |
| model: | |
| mask_feature_dim: 64 | |
| motion_dim: 512 | |
| rigs_dim: 169 | |
| tcn_config: | |
| depth: 6 | |
| dropout: 0.1 | |
| hidden: 1024 | |
| train: | |
| batch_size: 16 | |
| brow_boost_factor: 2.0 | |
| enable_brow_boost: true | |
| learning_rate: 0.0001 | |
| log_interval: 50 | |
| max_grad_norm: 1.0 | |
| mixed_precision: fp16 | |
| num_epochs: 100 | |
| output_dir: /home/deepspeed/workdir/FaceCapture/outputs/motion_mask_rig_weighted_2025-12-04_17-43-32 | |
| symmetric_weight: 1.0 | |
| use_scheduler: true | |
| use_symmetric_loss: true | |
| weight_decay: 0.0001 | |