| { |
| "config": { |
| "character_name": "pocoyo", |
| "module_name": "seq2seq", |
| "model": { |
| "hidden_size": 512, |
| "num_layers": 2, |
| "dropout": 0.3003, |
| "last_layer_mode": "ais" |
| }, |
| "masking": { |
| "mask_generator": { |
| "strategy_name": "random_block_keyframes", |
| "ratio_masked_block_keyframes": 0.0 |
| }, |
| "mask_applier": { |
| "concat_mode": "concat_dim", |
| "fill_mode": "zeros" |
| }, |
| "test_mask_generator": { |
| "strategy_name": "random_block_keyframes", |
| "ratio_masked_block_keyframes": 0.0 |
| }, |
| "predict_mask_generator": { |
| "strategy_name": "manual" |
| } |
| }, |
| "optimization": { |
| "optimizer": { |
| "name": "AdamW", |
| "lr": 0.0001, |
| "tune_lr": false, |
| "params": { |
| "weight_decay": 0.001 |
| } |
| }, |
| "lr_scheduler": { |
| "name": "CyclicLR", |
| "params": { |
| "base_lr": 6.897e-07, |
| "max_lr": 0.001805, |
| "step_size_up": 100.0, |
| "mode": "triangular2" |
| } |
| }, |
| "regularization": { |
| "last_layer_regularization": false, |
| "last_layer_regularization_weight": 0.001, |
| "full_model_regularization": false, |
| "full_model_regularization_weight": 1e-08 |
| } |
| }, |
| "losses": { |
| "pose_loss": { |
| "reduction": "none" |
| }, |
| "speed_loss": { |
| "reduction": "mean" |
| }, |
| "acceleration_loss": { |
| "reduction": "mean" |
| }, |
| "jerk_loss": { |
| "reduction": "mean" |
| }, |
| "controller_keyframes_prediction_loss": { |
| "reduction": "none", |
| "pos_weight": 1.0 |
| }, |
| "weights": { |
| "pose": 1.0, |
| "speed": 0.0, |
| "acceleration": 0.0, |
| "jerk": 0.0, |
| "controller_keyframes_prediction": 0.0, |
| "frame_weights": null, |
| "controller_keyframes_weights": null |
| }, |
| "auto_weighting_loss": true |
| }, |
| "do_controller_keyframes_prediction": false, |
| "post_processing": { |
| "do_controller_keyframes_reduction": false, |
| "threshold_controller_keyframes_reduction": null |
| } |
| }, |
| "controllers_config": { |
| "default_controllers_path": "motion_inbetweening/infra/configs/controllers_configs/pocoyo/pcy_default_controllers.yaml", |
| "trainable_controllers_path": "motion_inbetweening/infra/configs/controllers_configs/pocoyo/pcy_trainable_controllers_without_coeff.yaml", |
| "simplify_classes_path": null |
| }, |
| "transformations_keyframe_division_strategy": null, |
| "has_mean_pose_normalizer": false, |
| "has_std_pose_normalizer": false, |
| "epoch": 388, |
| "global_step": 19061 |
| } |
