Promote trial-076 (continuation from trial-075, +epochs 9-12): update trial_metrics.json

trial_metrics.json

@@ -1 +1 @@
-{"epoch": 8, "train_loss": 0.0701157002945741, "val_loss": 0.07329302156964938, "train_accuracy": 0.9752219321148825, "val_accuracy": 0.9739660574412533, "train_recall": 0.05376798287034035, "val_recall": 0.05109958350658417, "config": {"device": "cuda", "epochs": 8, "batch_size": 16, "learning_rate": 2e-05, "max_length": 1024, "model_name": "facebook/esm2_t33_650M_UR50D", "unfreeze_last_n": 27, "max_train_samples": 6000, "max_val_samples": 1000, "push_to_hub": true, "hf_repo_id": "kelmoir/cafa6-esm", "seed": 42, "output_dropout": 0.2, "esm_dropout": 0.0, "attention_dropout": 0.0, "grad_clip_norm": 1.0, "weight_decay": 0.2, "max_minutes": 55.0, "run_name": "trial-075", "trackio_space_id": null, "checkpoint_mode": "every-epoch", "checkpoint_branch": "trial-075", "hf_token_file": null, "resume_from_revision": null, "start_epoch": 0, "lr_schedule": "constant", "warmup_steps": null, "label_smoothing": 0.0, "decouple_bias_wd": false, "grad_accum_steps": 1, "pos_weight": false, "pos_weight_exponent": 0.5, "quiet": true, "use_lora": false, "lora_r": 16, "lora_alpha": 32, "lora_dropout": 0.05, "lora_target_modules": null, "lora_layers_to_transform": null, "mixed_precision": "bf16", "gradient_checkpointing": false, "run_diagnostic": true, "diagnostic_max_val_samples": 2000}}
+{"epoch": 12, "train_loss": 0.06569691857198874, "val_loss": 0.07257112249732017, "train_accuracy": 0.9766692776327242, "val_accuracy": 0.9739451697127938, "train_recall": 0.06732058525085449, "val_recall": 0.058190230280160904, "config": {"device": "cuda", "epochs": 4, "batch_size": 8, "learning_rate": 2e-05, "max_length": 1024, "model_name": "facebook/esm2_t33_650M_UR50D", "unfreeze_last_n": 27, "max_train_samples": 6000, "max_val_samples": 1000, "push_to_hub": false, "hf_repo_id": "kelmoir/cafa6-esm", "seed": 42, "output_dropout": 0.2, "esm_dropout": 0.0, "attention_dropout": 0.0, "grad_clip_norm": 1.0, "weight_decay": 0.2, "max_minutes": 45.0, "run_name": "trial-076", "trackio_space_id": null, "checkpoint_mode": "every-epoch", "checkpoint_branch": "trial-076", "hf_token_file": null, "resume_from_revision": "trial-075", "start_epoch": 8, "lr_schedule": "constant", "warmup_steps": null, "label_smoothing": 0.0, "decouple_bias_wd": false, "grad_accum_steps": 2, "pos_weight": false, "pos_weight_exponent": 0.5, "quiet": true, "use_lora": false, "lora_r": 16, "lora_alpha": 32, "lora_dropout": 0.05, "lora_target_modules": null, "lora_layers_to_transform": null, "mixed_precision": "bf16", "gradient_checkpointing": false, "run_diagnostic": true, "diagnostic_max_val_samples": 2000}}