| [ | |
| { | |
| "per_device_train_batch_size": 64, | |
| "gradient_accumulation_steps": 1, | |
| "per_device_eval_batch_size": 32, | |
| "eval_accumulation_steps": 1, | |
| "evaluation_strategy": "steps", | |
| "max_steps": 1000, | |
| "save_steps": 100, | |
| "eval_steps": 100, | |
| "logging_steps": 50, | |
| "learning_rate": 0.0001, | |
| "weight_decay": 0, | |
| "warmup_steps": 600, | |
| "save_total_limit": 1, | |
| "metric_for_best_model": "loss", | |
| "greater_is_better": false | |
| }, | |
| { | |
| "model_path": "facebook/wav2vec2-large-lv60", | |
| "pretrained_path": "/share/nas167/fuann/capt/pretrained-model/timit-wav2vec2-large-lv60-phoneme", | |
| "task_type": "mdd", | |
| "problem_type": "single_label_classification", | |
| "model_type": "baseline", | |
| "use_prompt": true, | |
| "final_dropout": 0.0, | |
| "ctc_zero_infinity": true, | |
| "layerdrop": 0.0, | |
| "activation_dropout": 0.0, | |
| "mask_time_prob": 0.095, | |
| "mask_time_length": 10, | |
| "mask_feature_prob": 1e-05, | |
| "mask_feature_length": 64 | |
| } | |
| ] |