ckpt_debug / checkpoint-10 /trainer_state.json
andyjzhao's picture
Upload folder using huggingface_hub
3d7cd17 verified
{
"best_global_step": 10,
"best_metric": 2.1194747907582294e+17,
"best_model_checkpoint": "/network/scratch/j/jianan.zhao/DNAFM/output/gencode_human_12.8k_12800/CKPT_DEBUG/checkpoint-10",
"epoch": 0.0014184397163120568,
"eval_steps": 10,
"global_step": 10,
"is_hyper_param_search": false,
"is_local_process_zero": true,
"is_world_process_zero": true,
"log_history": [
{
"comp/rl_weight": 0.03,
"comp/strictness": 0.0,
"epoch": 0.0014184397163120568,
"grad_norm": 917.1181030273438,
"loss": 144.7748,
"loss_ce": 54.116615295410156,
"loss_region": 0.10434026271104813,
"loss_total": 54.22095489501953,
"lr": 2.20454076850486e-05,
"router/selected_tokens_s0": 22.84375,
"router/selected_tokens_s1": 1.0,
"step": 10,
"tokens_trained": 0.01638272
},
{
"epoch": 0.0014184397163120568,
"eval_ppl": 2.1194747907582294e+17,
"eval_runtime": 1.7718,
"step": 10,
"tokens_trained": 0.01638272
},
{
"epoch": 0.0014184397163120568,
"eval_F": 7.813110399249941e-05,
"eval_F_cds": 2.439649178448139e-05,
"eval_F_dig": 7.813110399249941e-05,
"eval_F_exon": 6.169412055031155e-05,
"eval_F_intron": 8.631567698207511e-05,
"eval_F_nig": 8.867784676802712e-05,
"eval_F_promoter": 2.898993785611503e-05,
"eval_F_utr": 0.00013418766144453016,
"eval_G": 0.00022184538364639257,
"eval_G_cds": 0.0001901449227851035,
"eval_G_dig": 0.00015412581061020393,
"eval_G_exon": 0.000246234248719847,
"eval_G_intron": 0.00021249317768522492,
"eval_G_nig": 0.00019619516090846272,
"eval_G_promoter": 0.00030238687665044985,
"eval_G_utr": 0.0002784000847059613,
"eval_avg_bp_per_token": 12799.0,
"eval_bp_per_token/cds": 40989.5,
"eval_bp_per_token/dig": 12799.0,
"eval_bp_per_token/exon": 16209.0,
"eval_bp_per_token/intron": 11585.380952380952,
"eval_bp_per_token/nig": 11276.77358490566,
"eval_bp_per_token/promoter": 34494.72727272727,
"eval_bp_per_token/utr": 7452.25,
"eval_ppl_cds": 3.730425344687072e+19,
"eval_ppl_dig": 4.017993685449341e+17,
"eval_ppl_exon": 6.054858771331785e+16,
"eval_ppl_intron": 1.2271560232082048e+17,
"eval_ppl_nig": 3.849790568976093e+16,
"eval_ppl_promoter": 9.761527176325515e+19,
"eval_ppl_utr": 1.747612158170368e+17,
"step": 10,
"tokens_trained": 0.01638272
}
],
"logging_steps": 10,
"max_steps": 20,
"num_input_tokens_seen": 0,
"num_train_epochs": 1,
"save_steps": 10,
"stateful_callbacks": {
"TrainerControl": {
"args": {
"should_epoch_stop": false,
"should_evaluate": false,
"should_log": false,
"should_save": true,
"should_training_stop": false
},
"attributes": {}
}
},
"total_flos": 0.0,
"train_batch_size": 32,
"trial_name": null,
"trial_params": null
}