Add trial_metrics.json (carried over from trial-075) to complete the promoted checkpoint
Browse files- trial_metrics.json +1 -0
trial_metrics.json
ADDED
|
@@ -0,0 +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}}
|