| { |
| "repo_id": "BDRC/gyuyig-tsugdri-binary-script-classifier", |
| "train_dataset_id": "BDRC/gyuyig-tsugdri-binary-balanced-script-classification-dataset", |
| "warmstart_repo": "BDRC/4-class-balanced-script-classifier", |
| "task": "gyuyig_tsugdri_binary_classification", |
| "experiment": "dinov3_gyuyig_tsugdri_binary", |
| "classes": [ |
| "Gyuyig", |
| "Tsugdri" |
| ], |
| "pooling": "cls_token", |
| "preprocess": { |
| "train": "center_crop", |
| "val": "center_crop", |
| "test": "center_crop", |
| "size": 224 |
| }, |
| "warmstart": { |
| "warmstart_repo": "BDRC/4-class-balanced-script-classifier", |
| "warmstart_checkpoint": null, |
| "warmstart_checkpoint_file": "final_model.pt", |
| "checkpoint_test_metrics": { |
| "loss": 0.42663901050885517, |
| "accuracy": 0.9208333333333333, |
| "macro_f1": 0.9208489161207983, |
| "weighted_f1": 0.9208489161207983 |
| }, |
| "warmstart_pooling": "cls_token" |
| }, |
| "training_config": { |
| "lr_head_a": 0.0005, |
| "lr_backbone_b": 1e-05, |
| "lr_head_b": 0.0001, |
| "lr_backbone_c": 1.5e-05, |
| "lr_head_c": 5e-05, |
| "epochs_a": 7, |
| "epochs_b": 10, |
| "epochs_c": 12, |
| "unfreeze_blocks_b": 4, |
| "unfreeze_blocks_c": 8, |
| "skip_stage_c": false, |
| "scheduler": "cosine_warmup", |
| "warmup_epochs_a": 0, |
| "warmup_epochs_b": 1, |
| "warmup_epochs_c": 1, |
| "warmup_start_factor": 0.01, |
| "min_lr_ratio": 0.01, |
| "weight_decay": 0.02, |
| "grad_clip_norm": 1.0, |
| "class_weight_mode": "custom", |
| "class_weights": [ |
| 1.0, |
| 1.0 |
| ], |
| "label_smoothing": 0.05, |
| "dropout": 0.1, |
| "early_stop_patience": 6, |
| "early_stop_min_delta": 0.001, |
| "resolved_class_weights": [ |
| 1.0, |
| 1.0 |
| ], |
| "pooling": "cls_token", |
| "lambda_consistency": 0.0, |
| "flip_op": "rot180" |
| }, |
| "best_checkpoint": { |
| "path": "best_stage_c_last_blocks.pt", |
| "epoch": 1, |
| "val_macro_f1": 0.9164578111946533 |
| }, |
| "val_metrics": { |
| "loss": 0.3914561231931051, |
| "accuracy": 0.9166666666666666, |
| "macro_f1": 0.9164578111946533, |
| "weighted_f1": 0.9164578111946533, |
| "auc_roc": 0.9311111111111111 |
| }, |
| "test_metrics": { |
| "loss": 0.40474860469500223, |
| "accuracy": 0.85, |
| "macro_f1": 0.847930160518164, |
| "weighted_f1": 0.847930160518164, |
| "auc_roc": 0.9297222222222223 |
| } |
| } |
|
|
