| { | |
| "args": { | |
| "model_name": null, | |
| "encoder_model_name": "answerdotai/ModernBERT-base", | |
| "audio_model_name": "facebook/wav2vec2-base-960h", | |
| "save_path": "models/emo-gliclass-audio-bi-1-wds-0.015-red-sum", | |
| "data_path": "datasets/processed_dataset.json", | |
| "problem_type": "multi_label_classification", | |
| "pooler_type": "first", | |
| "scorer_type": "simple", | |
| "architecture_type": "audio-bi-encoder", | |
| "normalize_features": true, | |
| "extract_text_features": false, | |
| "prompt_first": true, | |
| "use_lstm": false, | |
| "squeeze_layers": false, | |
| "shuffle_labels": true, | |
| "num_epochs": 1, | |
| "batch_size": 8, | |
| "encoder_lr": 1e-05, | |
| "others_lr": 1e-05, | |
| "encoder_weight_decay": 0.015, | |
| "others_weight_decay": 0.015, | |
| "warmup_ratio": 0.05, | |
| "lr_scheduler_type": "cosine", | |
| "focal_loss_alpha": 0.9, | |
| "focal_loss_gamma": 2.5, | |
| "contrastive_loss_coef": 0.0, | |
| "max_length": 1024, | |
| "save_steps": 720, | |
| "save_total_limit": 3, | |
| "num_workers": 12, | |
| "fp16": false | |
| }, | |
| "eval_metrics": { | |
| "eval_loss": 1.1119344234466553, | |
| "eval_accuracy": 0.85234375, | |
| "eval_precision": 0.8347429327943167, | |
| "eval_recall": 0.85234375, | |
| "eval_f1": 0.8413624188086402, | |
| "eval_runtime": 15.7319, | |
| "eval_samples_per_second": 81.363, | |
| "eval_steps_per_second": 10.17, | |
| "epoch": 1.0 | |
| } | |
| } |