File size: 2,649 Bytes
64dc6ae | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 | {
"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
} |