| { |
| "best_val_war": 0.9436127744510978, |
| "best_val_uar": 0.8422065711699508, |
| "best_val_loss": 0.16450884051903517, |
| "best_epoch": 28, |
| "epoch_selection": { |
| "rule_id": "set_rule_war_then_uar_v1", |
| "rule": "retain every candidate whose validation WAR is within 1e-4 of the global maximum; within that set retain every candidate whose validation UAR is within 1e-4 of the retained maximum; within that, every candidate whose validation loss is within 1e-4 of the retained minimum; break remaining ties with the declared ordered keys. Order-independent by construction: no pairwise tolerance comparison is made.", |
| "tolerance": 0.0001, |
| "tie_breakers": [ |
| "epoch" |
| ], |
| "n_candidates": 120, |
| "max_val_war": 0.9436127744510978, |
| "n_retained_by_war": 1, |
| "max_val_uar_in_retained": 0.8422065711699508, |
| "n_retained_by_uar": 1, |
| "min_val_loss_in_retained": 0.16450884051903517, |
| "n_retained_by_loss": 1, |
| "selected": { |
| "epoch": 28, |
| "train_loss": 0.1893969083036129, |
| "val_war": 0.9436127744510978, |
| "val_uar": 0.8422065711699508, |
| "val_loss": 0.16450884051903517, |
| "val_weighted_f1": 0.9431165336357188, |
| "updates": 552 |
| } |
| }, |
| "epochs_completed": 120, |
| "n_retained_max": 2, |
| "retained_epochs": [ |
| 28 |
| ], |
| "params": { |
| "total": 85803270, |
| "trainable": 85803270, |
| "trainable_pct": 100.0 |
| }, |
| "param_groups": { |
| "decay": { |
| "tensors": 50, |
| "params": 85529088 |
| }, |
| "no_decay": { |
| "tensors": 102, |
| "params": 274182 |
| }, |
| "weight_decay": 0.05, |
| "rule": "architecture no_weight_decay() list, plus every tensor of rank <= 1", |
| "model_skip_list": [ |
| "cls_token", |
| "dist_token", |
| "pos_embed" |
| ], |
| "frozen_tensors": 0, |
| "no_decay_names": [ |
| "cls_token", |
| "pos_embed", |
| "patch_embed.proj.bias", |
| "blocks.0.norm1.weight", |
| "blocks.0.norm1.bias", |
| "blocks.0.attn.qkv.bias", |
| "blocks.0.attn.proj.bias", |
| "blocks.0.norm2.weight", |
| "blocks.0.norm2.bias", |
| "blocks.0.mlp.fc1.bias", |
| "blocks.0.mlp.fc2.bias", |
| "blocks.1.norm1.weight", |
| "blocks.1.norm1.bias", |
| "blocks.1.attn.qkv.bias", |
| "blocks.1.attn.proj.bias", |
| "blocks.1.norm2.weight", |
| "blocks.1.norm2.bias", |
| "blocks.1.mlp.fc1.bias", |
| "blocks.1.mlp.fc2.bias", |
| "blocks.2.norm1.weight", |
| "blocks.2.norm1.bias", |
| "blocks.2.attn.qkv.bias", |
| "blocks.2.attn.proj.bias", |
| "blocks.2.norm2.weight", |
| "blocks.2.norm2.bias", |
| "blocks.2.mlp.fc1.bias", |
| "blocks.2.mlp.fc2.bias", |
| "blocks.3.norm1.weight", |
| "blocks.3.norm1.bias", |
| "blocks.3.attn.qkv.bias", |
| "blocks.3.attn.proj.bias", |
| "blocks.3.norm2.weight", |
| "blocks.3.norm2.bias", |
| "blocks.3.mlp.fc1.bias", |
| "blocks.3.mlp.fc2.bias", |
| "blocks.4.norm1.weight", |
| "blocks.4.norm1.bias", |
| "blocks.4.attn.qkv.bias", |
| "blocks.4.attn.proj.bias", |
| "blocks.4.norm2.weight", |
| "blocks.4.norm2.bias", |
| "blocks.4.mlp.fc1.bias", |
| "blocks.4.mlp.fc2.bias", |
| "blocks.5.norm1.weight", |
| "blocks.5.norm1.bias", |
| "blocks.5.attn.qkv.bias", |
| "blocks.5.attn.proj.bias", |
| "blocks.5.norm2.weight", |
| "blocks.5.norm2.bias", |
| "blocks.5.mlp.fc1.bias", |
| "blocks.5.mlp.fc2.bias", |
| "blocks.6.norm1.weight", |
| "blocks.6.norm1.bias", |
| "blocks.6.attn.qkv.bias", |
| "blocks.6.attn.proj.bias", |
| "blocks.6.norm2.weight", |
| "blocks.6.norm2.bias", |
| "blocks.6.mlp.fc1.bias", |
| "blocks.6.mlp.fc2.bias", |
| "blocks.7.norm1.weight", |
| "blocks.7.norm1.bias", |
| "blocks.7.attn.qkv.bias", |
| "blocks.7.attn.proj.bias", |
| "blocks.7.norm2.weight", |
| "blocks.7.norm2.bias", |
| "blocks.7.mlp.fc1.bias", |
| "blocks.7.mlp.fc2.bias", |
| "blocks.8.norm1.weight", |
| "blocks.8.norm1.bias", |
| "blocks.8.attn.qkv.bias", |
| "blocks.8.attn.proj.bias", |
| "blocks.8.norm2.weight", |
| "blocks.8.norm2.bias", |
| "blocks.8.mlp.fc1.bias", |
| "blocks.8.mlp.fc2.bias", |
| "blocks.9.norm1.weight", |
| "blocks.9.norm1.bias", |
| "blocks.9.attn.qkv.bias", |
| "blocks.9.attn.proj.bias", |
| "blocks.9.norm2.weight", |
| "blocks.9.norm2.bias", |
| "blocks.9.mlp.fc1.bias", |
| "blocks.9.mlp.fc2.bias", |
| "blocks.10.norm1.weight", |
| "blocks.10.norm1.bias", |
| "blocks.10.attn.qkv.bias", |
| "blocks.10.attn.proj.bias", |
| "blocks.10.norm2.weight", |
| "blocks.10.norm2.bias", |
| "blocks.10.mlp.fc1.bias", |
| "blocks.10.mlp.fc2.bias", |
| "blocks.11.norm1.weight", |
| "blocks.11.norm1.bias", |
| "blocks.11.attn.qkv.bias", |
| "blocks.11.attn.proj.bias", |
| "blocks.11.norm2.weight", |
| "blocks.11.norm2.bias", |
| "blocks.11.mlp.fc1.bias", |
| "blocks.11.mlp.fc2.bias", |
| "norm.weight", |
| "norm.bias", |
| "head.bias" |
| ], |
| "total_steps": 66240, |
| "warmup_steps": 1104 |
| }, |
| "run_identity": { |
| "method": "full_ft", |
| "protocol": "official_split", |
| "split_seed": 42, |
| "model_seed": 42, |
| "source_stage": "augreg", |
| "init_from_sha256": null, |
| "data_root": "data/ferplus" |
| }, |
| "checkpoint_sha256": "3094f103c17bd13558f960c22b91ed3316679cadeab88ba269d862019c4dd58a", |
| "amp_active": true, |
| "train_time_sec": 18801.249953985214, |
| "time_per_epoch_sec": 156.6770829498768, |
| "peak_mem_gb": 3.2119665145874023, |
| "skipped_steps": 28, |
| "optimizer_updates": 66212, |
| "history": [ |
| { |
| "epoch": 1, |
| "train_loss": 1.4047704333016626, |
| "val_war": 0.8223552894211577, |
| "val_uar": 0.6159856467505688, |
| "val_loss": 0.5097239920479095, |
| "val_weighted_f1": 0.8197161181996515, |
| "updates": 547 |
| }, |
| { |
| "epoch": 2, |
| "train_loss": 0.6416764952992455, |
| "val_war": 0.8837325349301397, |
| "val_uar": 0.651311836222769, |
| "val_loss": 0.3376162086061375, |
| "val_weighted_f1": 0.8729204991746009, |
| "updates": 552 |
| }, |
| { |
| "epoch": 3, |
| "train_loss": 0.4902889822945771, |
| "val_war": 0.9106786427145709, |
| "val_uar": 0.7028063560588943, |
| "val_loss": 0.2467307984650492, |
| "val_weighted_f1": 0.9039909599003104, |
| "updates": 552 |
| }, |
| { |
| "epoch": 4, |
| "train_loss": 0.41320013329804595, |
| "val_war": 0.9101796407185628, |
| "val_uar": 0.758151098367541, |
| "val_loss": 0.23912653208016874, |
| "val_weighted_f1": 0.9078919964997306, |
| "updates": 552 |
| }, |
| { |
| "epoch": 5, |
| "train_loss": 0.3782459644850952, |
| "val_war": 0.9276447105788423, |
| "val_uar": 0.7527450839795714, |
| "val_loss": 0.19767410455349677, |
| "val_weighted_f1": 0.9233663934684276, |
| "updates": 552 |
| }, |
| { |
| "epoch": 6, |
| "train_loss": 0.3450085104919954, |
| "val_war": 0.9316367265469062, |
| "val_uar": 0.774065657914138, |
| "val_loss": 0.19791446635109222, |
| "val_weighted_f1": 0.9285206478898584, |
| "updates": 552 |
| }, |
| { |
| "epoch": 7, |
| "train_loss": 0.3270069262146119, |
| "val_war": 0.9336327345309381, |
| "val_uar": 0.7944838654815302, |
| "val_loss": 0.1838007161687472, |
| "val_weighted_f1": 0.9319617105616057, |
| "updates": 552 |
| }, |
| { |
| "epoch": 8, |
| "train_loss": 0.3104032961829284, |
| "val_war": 0.9266467065868264, |
| "val_uar": 0.7657295042459896, |
| "val_loss": 0.19788938168874995, |
| "val_weighted_f1": 0.9232278845048117, |
| "updates": 552 |
| }, |
| { |
| "epoch": 9, |
| "train_loss": 0.29067675521806524, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8348326702929394, |
| "val_loss": 0.18219371157848907, |
| "val_weighted_f1": 0.9392222992123364, |
| "updates": 552 |
| }, |
| { |
| "epoch": 10, |
| "train_loss": 0.28042608418145076, |
| "val_war": 0.9336327345309381, |
| "val_uar": 0.82131829483203, |
| "val_loss": 0.177239563234314, |
| "val_weighted_f1": 0.931834378555007, |
| "updates": 552 |
| }, |
| { |
| "epoch": 11, |
| "train_loss": 0.262686960672253, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8038431213620192, |
| "val_loss": 0.17341571445474604, |
| "val_weighted_f1": 0.9376495320877638, |
| "updates": 552 |
| }, |
| { |
| "epoch": 12, |
| "train_loss": 0.26096923651759274, |
| "val_war": 0.9356287425149701, |
| "val_uar": 0.8421289800253589, |
| "val_loss": 0.1788175468316335, |
| "val_weighted_f1": 0.9353055837721055, |
| "updates": 552 |
| }, |
| { |
| "epoch": 13, |
| "train_loss": 0.25120533699118364, |
| "val_war": 0.9391217564870259, |
| "val_uar": 0.8051524911546393, |
| "val_loss": 0.1654306911601278, |
| "val_weighted_f1": 0.9375012220426306, |
| "updates": 552 |
| }, |
| { |
| "epoch": 14, |
| "train_loss": 0.24476321863254186, |
| "val_war": 0.9406187624750499, |
| "val_uar": 0.8326409383075886, |
| "val_loss": 0.167763271701788, |
| "val_weighted_f1": 0.9392802550828819, |
| "updates": 552 |
| }, |
| { |
| "epoch": 15, |
| "train_loss": 0.237078241127205, |
| "val_war": 0.936127744510978, |
| "val_uar": 0.8122128463652457, |
| "val_loss": 0.17576297711647912, |
| "val_weighted_f1": 0.9343963572417492, |
| "updates": 551 |
| }, |
| { |
| "epoch": 16, |
| "train_loss": 0.22895441080083875, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8150744750602922, |
| "val_loss": 0.17539126556850004, |
| "val_weighted_f1": 0.9359240875153829, |
| "updates": 552 |
| }, |
| { |
| "epoch": 17, |
| "train_loss": 0.22731330655302032, |
| "val_war": 0.9346307385229541, |
| "val_uar": 0.8057130695171434, |
| "val_loss": 0.17710513801512842, |
| "val_weighted_f1": 0.9327008395248881, |
| "updates": 552 |
| }, |
| { |
| "epoch": 18, |
| "train_loss": 0.21861273906168927, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8428411018338874, |
| "val_loss": 0.16258980455750716, |
| "val_weighted_f1": 0.9387422093235203, |
| "updates": 552 |
| }, |
| { |
| "epoch": 19, |
| "train_loss": 0.2129864644221858, |
| "val_war": 0.9406187624750499, |
| "val_uar": 0.8102733595939228, |
| "val_loss": 0.16628249388136074, |
| "val_weighted_f1": 0.9389083202499009, |
| "updates": 551 |
| }, |
| { |
| "epoch": 20, |
| "train_loss": 0.21217662969697104, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8278308407678868, |
| "val_loss": 0.16899946621970502, |
| "val_weighted_f1": 0.9374319633540159, |
| "updates": 552 |
| }, |
| { |
| "epoch": 21, |
| "train_loss": 0.20610570450874843, |
| "val_war": 0.9341317365269461, |
| "val_uar": 0.8014836738025538, |
| "val_loss": 0.17031347205836378, |
| "val_weighted_f1": 0.9324509759191477, |
| "updates": 552 |
| }, |
| { |
| "epoch": 22, |
| "train_loss": 0.2021111387766412, |
| "val_war": 0.937125748502994, |
| "val_uar": 0.8233250264680203, |
| "val_loss": 0.17300484442484831, |
| "val_weighted_f1": 0.9356376442336026, |
| "updates": 552 |
| }, |
| { |
| "epoch": 23, |
| "train_loss": 0.20047183184811201, |
| "val_war": 0.9416167664670658, |
| "val_uar": 0.8327062078589332, |
| "val_loss": 0.1606178121444232, |
| "val_weighted_f1": 0.9403728956253352, |
| "updates": 551 |
| }, |
| { |
| "epoch": 24, |
| "train_loss": 0.19719777391957236, |
| "val_war": 0.9421157684630739, |
| "val_uar": 0.842356635721179, |
| "val_loss": 0.1603443944852747, |
| "val_weighted_f1": 0.9416712024861985, |
| "updates": 552 |
| }, |
| { |
| "epoch": 25, |
| "train_loss": 0.19344528113726578, |
| "val_war": 0.935129740518962, |
| "val_uar": 0.8543851100798032, |
| "val_loss": 0.17777673513470535, |
| "val_weighted_f1": 0.9348537743649601, |
| "updates": 552 |
| }, |
| { |
| "epoch": 26, |
| "train_loss": 0.1901420341960486, |
| "val_war": 0.9331337325349301, |
| "val_uar": 0.7908487120788793, |
| "val_loss": 0.17135679430352477, |
| "val_weighted_f1": 0.930950739360817, |
| "updates": 552 |
| }, |
| { |
| "epoch": 27, |
| "train_loss": 0.19049885533790156, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8379346014513839, |
| "val_loss": 0.1668140509736514, |
| "val_weighted_f1": 0.9383352279026699, |
| "updates": 551 |
| }, |
| { |
| "epoch": 28, |
| "train_loss": 0.1893969083036129, |
| "val_war": 0.9436127744510978, |
| "val_uar": 0.8422065711699508, |
| "val_loss": 0.16450884051903517, |
| "val_weighted_f1": 0.9431165336357188, |
| "updates": 552 |
| }, |
| { |
| "epoch": 29, |
| "train_loss": 0.1831858852728049, |
| "val_war": 0.9311377245508982, |
| "val_uar": 0.8308813161602101, |
| "val_loss": 0.19041404312480234, |
| "val_weighted_f1": 0.9299362619686248, |
| "updates": 552 |
| }, |
| { |
| "epoch": 30, |
| "train_loss": 0.18308283137377582, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8314178853183338, |
| "val_loss": 0.16331172422199192, |
| "val_weighted_f1": 0.93878645372026, |
| "updates": 552 |
| }, |
| { |
| "epoch": 31, |
| "train_loss": 0.18197595729050398, |
| "val_war": 0.9321357285429142, |
| "val_uar": 0.7967183493197861, |
| "val_loss": 0.17413552166399604, |
| "val_weighted_f1": 0.929998871426466, |
| "updates": 552 |
| }, |
| { |
| "epoch": 32, |
| "train_loss": 0.17810741992778437, |
| "val_war": 0.9356287425149701, |
| "val_uar": 0.807824984251555, |
| "val_loss": 0.1836102489166393, |
| "val_weighted_f1": 0.9332500180606241, |
| "updates": 552 |
| }, |
| { |
| "epoch": 33, |
| "train_loss": 0.17732757957780215, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8359269027357614, |
| "val_loss": 0.1672114410145792, |
| "val_weighted_f1": 0.9373836502541006, |
| "updates": 552 |
| }, |
| { |
| "epoch": 34, |
| "train_loss": 0.17661268946852782, |
| "val_war": 0.9346307385229541, |
| "val_uar": 0.8154203544688675, |
| "val_loss": 0.1725583569910712, |
| "val_weighted_f1": 0.93333122447792, |
| "updates": 551 |
| }, |
| { |
| "epoch": 35, |
| "train_loss": 0.17519370785877797, |
| "val_war": 0.937125748502994, |
| "val_uar": 0.8297162145398028, |
| "val_loss": 0.17537833521228113, |
| "val_weighted_f1": 0.9363643266716352, |
| "updates": 552 |
| }, |
| { |
| "epoch": 36, |
| "train_loss": 0.1711410973390961, |
| "val_war": 0.936127744510978, |
| "val_uar": 0.8217137410289959, |
| "val_loss": 0.17358410074384864, |
| "val_weighted_f1": 0.9349008181696952, |
| "updates": 552 |
| }, |
| { |
| "epoch": 37, |
| "train_loss": 0.17219136656428133, |
| "val_war": 0.9411177644710579, |
| "val_uar": 0.8270144419891348, |
| "val_loss": 0.16602112324592358, |
| "val_weighted_f1": 0.9395836841972692, |
| "updates": 552 |
| }, |
| { |
| "epoch": 38, |
| "train_loss": 0.16979345952638533, |
| "val_war": 0.9236526946107785, |
| "val_uar": 0.8008375266113957, |
| "val_loss": 0.18842854220680372, |
| "val_weighted_f1": 0.9227581205840718, |
| "updates": 550 |
| }, |
| { |
| "epoch": 39, |
| "train_loss": 0.16958456639603348, |
| "val_war": 0.9346307385229541, |
| "val_uar": 0.8113874086070426, |
| "val_loss": 0.18208533474904096, |
| "val_weighted_f1": 0.9329059251789252, |
| "updates": 552 |
| }, |
| { |
| "epoch": 40, |
| "train_loss": 0.1659080766187424, |
| "val_war": 0.9331337325349301, |
| "val_uar": 0.8158811814418448, |
| "val_loss": 0.18062037025799532, |
| "val_weighted_f1": 0.9316418452003805, |
| "updates": 552 |
| }, |
| { |
| "epoch": 41, |
| "train_loss": 0.16519805883967528, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8225274975838978, |
| "val_loss": 0.17126444376216915, |
| "val_weighted_f1": 0.936323868589868, |
| "updates": 552 |
| }, |
| { |
| "epoch": 42, |
| "train_loss": 0.1648560220337773, |
| "val_war": 0.9366267465069861, |
| "val_uar": 0.8340993713289312, |
| "val_loss": 0.1728722200749401, |
| "val_weighted_f1": 0.9355585459186554, |
| "updates": 552 |
| }, |
| { |
| "epoch": 43, |
| "train_loss": 0.16293427874502206, |
| "val_war": 0.9311377245508982, |
| "val_uar": 0.8229728037252396, |
| "val_loss": 0.18072660178124547, |
| "val_weighted_f1": 0.9299140972565592, |
| "updates": 552 |
| }, |
| { |
| "epoch": 44, |
| "train_loss": 0.16389999659551008, |
| "val_war": 0.9331337325349301, |
| "val_uar": 0.8378471805786791, |
| "val_loss": 0.17224340731214857, |
| "val_weighted_f1": 0.9322058935057838, |
| "updates": 552 |
| }, |
| { |
| "epoch": 45, |
| "train_loss": 0.161916443627807, |
| "val_war": 0.935129740518962, |
| "val_uar": 0.8314001098910526, |
| "val_loss": 0.17295021951912407, |
| "val_weighted_f1": 0.934366217306364, |
| "updates": 551 |
| }, |
| { |
| "epoch": 46, |
| "train_loss": 0.16076805402007321, |
| "val_war": 0.9291417165668663, |
| "val_uar": 0.8174550410960005, |
| "val_loss": 0.18424697141506952, |
| "val_weighted_f1": 0.9273992850694578, |
| "updates": 552 |
| }, |
| { |
| "epoch": 47, |
| "train_loss": 0.15887886772722531, |
| "val_war": 0.9346307385229541, |
| "val_uar": 0.8365652156390958, |
| "val_loss": 0.1765581264705239, |
| "val_weighted_f1": 0.9334462961038144, |
| "updates": 552 |
| }, |
| { |
| "epoch": 48, |
| "train_loss": 0.15737028310519563, |
| "val_war": 0.9336327345309381, |
| "val_uar": 0.8169661806681837, |
| "val_loss": 0.17476384503815226, |
| "val_weighted_f1": 0.9322262324323962, |
| "updates": 552 |
| }, |
| { |
| "epoch": 49, |
| "train_loss": 0.15786523003479178, |
| "val_war": 0.9406187624750499, |
| "val_uar": 0.8254044970943709, |
| "val_loss": 0.1667531690971104, |
| "val_weighted_f1": 0.939319809871151, |
| "updates": 551 |
| }, |
| { |
| "epoch": 50, |
| "train_loss": 0.1579624950213781, |
| "val_war": 0.936127744510978, |
| "val_uar": 0.8327343408522303, |
| "val_loss": 0.17719584872979605, |
| "val_weighted_f1": 0.9346604369692918, |
| "updates": 552 |
| }, |
| { |
| "epoch": 51, |
| "train_loss": 0.15547056047424304, |
| "val_war": 0.9426147704590818, |
| "val_uar": 0.8433774102687055, |
| "val_loss": 0.16802481837020425, |
| "val_weighted_f1": 0.9420444338655707, |
| "updates": 552 |
| }, |
| { |
| "epoch": 52, |
| "train_loss": 0.15682258752461337, |
| "val_war": 0.936127744510978, |
| "val_uar": 0.8350100725401344, |
| "val_loss": 0.16921452003086873, |
| "val_weighted_f1": 0.9350474432847485, |
| "updates": 552 |
| }, |
| { |
| "epoch": 53, |
| "train_loss": 0.15364923303130396, |
| "val_war": 0.9311377245508982, |
| "val_uar": 0.8073020137932175, |
| "val_loss": 0.18136151143890655, |
| "val_weighted_f1": 0.9296318489900549, |
| "updates": 551 |
| }, |
| { |
| "epoch": 54, |
| "train_loss": 0.15273373563254794, |
| "val_war": 0.9406187624750499, |
| "val_uar": 0.8307596572682553, |
| "val_loss": 0.16410571304029095, |
| "val_weighted_f1": 0.939278460281594, |
| "updates": 552 |
| }, |
| { |
| "epoch": 55, |
| "train_loss": 0.1523624458734621, |
| "val_war": 0.9331337325349301, |
| "val_uar": 0.8336914121804714, |
| "val_loss": 0.18340963726034185, |
| "val_weighted_f1": 0.9323174517499838, |
| "updates": 552 |
| }, |
| { |
| "epoch": 56, |
| "train_loss": 0.1521285242463271, |
| "val_war": 0.936127744510978, |
| "val_uar": 0.8597611370910458, |
| "val_loss": 0.171738973649557, |
| "val_weighted_f1": 0.9358443761140609, |
| "updates": 552 |
| }, |
| { |
| "epoch": 57, |
| "train_loss": 0.149947994185449, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8275117786242937, |
| "val_loss": 0.17215411481029258, |
| "val_weighted_f1": 0.9366187640703773, |
| "updates": 551 |
| }, |
| { |
| "epoch": 58, |
| "train_loss": 0.14973477020451964, |
| "val_war": 0.9356287425149701, |
| "val_uar": 0.8265770511627251, |
| "val_loss": 0.17521684984128155, |
| "val_weighted_f1": 0.9348368439643246, |
| "updates": 552 |
| }, |
| { |
| "epoch": 59, |
| "train_loss": 0.14835869101203336, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8121690458857072, |
| "val_loss": 0.17793412170486297, |
| "val_weighted_f1": 0.9377800098595549, |
| "updates": 552 |
| }, |
| { |
| "epoch": 60, |
| "train_loss": 0.14778099850205942, |
| "val_war": 0.937125748502994, |
| "val_uar": 0.8374458697734122, |
| "val_loss": 0.18030216605839378, |
| "val_weighted_f1": 0.9359710989521886, |
| "updates": 551 |
| }, |
| { |
| "epoch": 61, |
| "train_loss": 0.14852110827791745, |
| "val_war": 0.9366267465069861, |
| "val_uar": 0.8326849490326813, |
| "val_loss": 0.1811545911782278, |
| "val_weighted_f1": 0.9354929700443406, |
| "updates": 552 |
| }, |
| { |
| "epoch": 62, |
| "train_loss": 0.14789182282606472, |
| "val_war": 0.9411177644710579, |
| "val_uar": 0.8379392000499566, |
| "val_loss": 0.18187752142994704, |
| "val_weighted_f1": 0.9399631953367963, |
| "updates": 552 |
| }, |
| { |
| "epoch": 63, |
| "train_loss": 0.1469726145730114, |
| "val_war": 0.9321357285429142, |
| "val_uar": 0.8046150840463109, |
| "val_loss": 0.19261429753310666, |
| "val_weighted_f1": 0.930283583662995, |
| "updates": 552 |
| }, |
| { |
| "epoch": 64, |
| "train_loss": 0.14675609817076565, |
| "val_war": 0.93812375249501, |
| "val_uar": 0.8181238952943793, |
| "val_loss": 0.1784201398224174, |
| "val_weighted_f1": 0.9363240961445806, |
| "updates": 552 |
| }, |
| { |
| "epoch": 65, |
| "train_loss": 0.14619851021528663, |
| "val_war": 0.9401197604790419, |
| "val_uar": 0.8114451914341151, |
| "val_loss": 0.17476017050400466, |
| "val_weighted_f1": 0.9383031809394751, |
| "updates": 552 |
| }, |
| { |
| "epoch": 66, |
| "train_loss": 0.14603796090125956, |
| "val_war": 0.9366267465069861, |
| "val_uar": 0.8074483129594299, |
| "val_loss": 0.18011196317549, |
| "val_weighted_f1": 0.9348054209652275, |
| "updates": 552 |
| }, |
| { |
| "epoch": 67, |
| "train_loss": 0.14329448090157598, |
| "val_war": 0.9356287425149701, |
| "val_uar": 0.8106711128985123, |
| "val_loss": 0.1846027246968237, |
| "val_weighted_f1": 0.9337210658403128, |
| "updates": 552 |
| }, |
| { |
| "epoch": 68, |
| "train_loss": 0.1428006283184002, |
| "val_war": 0.9316367265469062, |
| "val_uar": 0.8031366712868886, |
| "val_loss": 0.18349386024439407, |
| "val_weighted_f1": 0.9300153443926755, |
| "updates": 551 |
| }, |
| { |
| "epoch": 69, |
| "train_loss": 0.14301485549754567, |
| "val_war": 0.9391217564870259, |
| "val_uar": 0.8263765855244181, |
| "val_loss": 0.17007501188390506, |
| "val_weighted_f1": 0.9379734352383235, |
| "updates": 552 |
| }, |
| { |
| "epoch": 70, |
| "train_loss": 0.1432912688790776, |
| "val_war": 0.9346307385229541, |
| "val_uar": 0.8111892343577506, |
| "val_loss": 0.18101948895259293, |
| "val_weighted_f1": 0.9330041838798201, |
| "updates": 552 |
| }, |
| { |
| "epoch": 71, |
| "train_loss": 0.14144405708899682, |
| "val_war": 0.93812375249501, |
| "val_uar": 0.8278699721986621, |
| "val_loss": 0.17523026980920703, |
| "val_weighted_f1": 0.9371255511915308, |
| "updates": 552 |
| }, |
| { |
| "epoch": 72, |
| "train_loss": 0.1411079731306863, |
| "val_war": 0.936127744510978, |
| "val_uar": 0.8390960814625954, |
| "val_loss": 0.17221887738345387, |
| "val_weighted_f1": 0.9357038867595254, |
| "updates": 551 |
| }, |
| { |
| "epoch": 73, |
| "train_loss": 0.141164897639828, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8268934771477933, |
| "val_loss": 0.17504457956897523, |
| "val_weighted_f1": 0.9361964607187647, |
| "updates": 552 |
| }, |
| { |
| "epoch": 74, |
| "train_loss": 0.1412509002880931, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8344505680864871, |
| "val_loss": 0.173524490224863, |
| "val_weighted_f1": 0.936921531468762, |
| "updates": 552 |
| }, |
| { |
| "epoch": 75, |
| "train_loss": 0.13983115238121133, |
| "val_war": 0.9346307385229541, |
| "val_uar": 0.8339003855987236, |
| "val_loss": 0.179867926739647, |
| "val_weighted_f1": 0.9336630157149802, |
| "updates": 552 |
| }, |
| { |
| "epoch": 76, |
| "train_loss": 0.14020412397073298, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8262726983698517, |
| "val_loss": 0.17286987647324978, |
| "val_weighted_f1": 0.9364524259053449, |
| "updates": 551 |
| }, |
| { |
| "epoch": 77, |
| "train_loss": 0.13932993502339913, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8414604555357621, |
| "val_loss": 0.17529048795947533, |
| "val_weighted_f1": 0.9386938672967305, |
| "updates": 552 |
| }, |
| { |
| "epoch": 78, |
| "train_loss": 0.13933016522991967, |
| "val_war": 0.937125748502994, |
| "val_uar": 0.8275055800868318, |
| "val_loss": 0.1820735025429678, |
| "val_weighted_f1": 0.9356624613816862, |
| "updates": 552 |
| }, |
| { |
| "epoch": 79, |
| "train_loss": 0.13874173161328518, |
| "val_war": 0.9411177644710579, |
| "val_uar": 0.830963406519445, |
| "val_loss": 0.1751293259972108, |
| "val_weighted_f1": 0.9400655519928833, |
| "updates": 551 |
| }, |
| { |
| "epoch": 80, |
| "train_loss": 0.13859787873970633, |
| "val_war": 0.9391217564870259, |
| "val_uar": 0.8451352078232186, |
| "val_loss": 0.17977165573549841, |
| "val_weighted_f1": 0.9384386366593882, |
| "updates": 552 |
| }, |
| { |
| "epoch": 81, |
| "train_loss": 0.13631444165598036, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.832240165663004, |
| "val_loss": 0.1811827164061531, |
| "val_weighted_f1": 0.938325838303113, |
| "updates": 552 |
| }, |
| { |
| "epoch": 82, |
| "train_loss": 0.13703395100576898, |
| "val_war": 0.9411177644710579, |
| "val_uar": 0.8304851210394378, |
| "val_loss": 0.1771898951656566, |
| "val_weighted_f1": 0.9401466011116297, |
| "updates": 552 |
| }, |
| { |
| "epoch": 83, |
| "train_loss": 0.13599931567773949, |
| "val_war": 0.936127744510978, |
| "val_uar": 0.8188222957698937, |
| "val_loss": 0.18242532733136307, |
| "val_weighted_f1": 0.9346576422860918, |
| "updates": 551 |
| }, |
| { |
| "epoch": 84, |
| "train_loss": 0.13594216169147516, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8317135786104174, |
| "val_loss": 0.1794372118132081, |
| "val_weighted_f1": 0.9381984900340161, |
| "updates": 552 |
| }, |
| { |
| "epoch": 85, |
| "train_loss": 0.13535929553944775, |
| "val_war": 0.9406187624750499, |
| "val_uar": 0.8263293364063524, |
| "val_loss": 0.17809502831000293, |
| "val_weighted_f1": 0.9391574440528745, |
| "updates": 552 |
| }, |
| { |
| "epoch": 86, |
| "train_loss": 0.13495322340902785, |
| "val_war": 0.9366267465069861, |
| "val_uar": 0.8284884049407508, |
| "val_loss": 0.18140700646860156, |
| "val_weighted_f1": 0.9353187792505459, |
| "updates": 552 |
| }, |
| { |
| "epoch": 87, |
| "train_loss": 0.1354656302687818, |
| "val_war": 0.9391217564870259, |
| "val_uar": 0.8247989486971029, |
| "val_loss": 0.1832676084931501, |
| "val_weighted_f1": 0.937619021017889, |
| "updates": 552 |
| }, |
| { |
| "epoch": 88, |
| "train_loss": 0.13511739565197464, |
| "val_war": 0.9391217564870259, |
| "val_uar": 0.8168166353390379, |
| "val_loss": 0.1806866105593845, |
| "val_weighted_f1": 0.937421845224609, |
| "updates": 551 |
| }, |
| { |
| "epoch": 89, |
| "train_loss": 0.13420768714540032, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8182509202713549, |
| "val_loss": 0.17914717357434676, |
| "val_weighted_f1": 0.9362079011897461, |
| "updates": 552 |
| }, |
| { |
| "epoch": 90, |
| "train_loss": 0.13419607986263038, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.820215573308591, |
| "val_loss": 0.18154250928563273, |
| "val_weighted_f1": 0.9380655094948194, |
| "updates": 552 |
| }, |
| { |
| "epoch": 91, |
| "train_loss": 0.133312442309829, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8252876864403995, |
| "val_loss": 0.182653937778787, |
| "val_weighted_f1": 0.93646878054044, |
| "updates": 552 |
| }, |
| { |
| "epoch": 92, |
| "train_loss": 0.13295968446162368, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8235300876715872, |
| "val_loss": 0.17950444788989905, |
| "val_weighted_f1": 0.9372398945588261, |
| "updates": 552 |
| }, |
| { |
| "epoch": 93, |
| "train_loss": 0.13368176660735737, |
| "val_war": 0.935129740518962, |
| "val_uar": 0.8109127337591948, |
| "val_loss": 0.18271107064511724, |
| "val_weighted_f1": 0.9332663929251748, |
| "updates": 552 |
| }, |
| { |
| "epoch": 94, |
| "train_loss": 0.13279324432373937, |
| "val_war": 0.9356287425149701, |
| "val_uar": 0.8206186113891706, |
| "val_loss": 0.17936357733851183, |
| "val_weighted_f1": 0.9343506117826228, |
| "updates": 552 |
| }, |
| { |
| "epoch": 95, |
| "train_loss": 0.13153805357087106, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8063027351163772, |
| "val_loss": 0.18261967840308915, |
| "val_weighted_f1": 0.9366194779986263, |
| "updates": 551 |
| }, |
| { |
| "epoch": 96, |
| "train_loss": 0.13274440862684572, |
| "val_war": 0.9341317365269461, |
| "val_uar": 0.8105302216756279, |
| "val_loss": 0.18330342991801793, |
| "val_weighted_f1": 0.932406482401329, |
| "updates": 552 |
| }, |
| { |
| "epoch": 97, |
| "train_loss": 0.1322247357060265, |
| "val_war": 0.93812375249501, |
| "val_uar": 0.8216714461193625, |
| "val_loss": 0.18368515761907467, |
| "val_weighted_f1": 0.936473096566982, |
| "updates": 552 |
| }, |
| { |
| "epoch": 98, |
| "train_loss": 0.13247279008922952, |
| "val_war": 0.9401197604790419, |
| "val_uar": 0.8232914701423778, |
| "val_loss": 0.1798601560815128, |
| "val_weighted_f1": 0.938497165276114, |
| "updates": 552 |
| }, |
| { |
| "epoch": 99, |
| "train_loss": 0.1319061137681392, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8274636360120335, |
| "val_loss": 0.17892851613595814, |
| "val_weighted_f1": 0.9371650037471614, |
| "updates": 551 |
| }, |
| { |
| "epoch": 100, |
| "train_loss": 0.13156906532623985, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8294923307976768, |
| "val_loss": 0.18129546410904196, |
| "val_weighted_f1": 0.9381280097928196, |
| "updates": 552 |
| }, |
| { |
| "epoch": 101, |
| "train_loss": 0.13197330744784788, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8283308906118463, |
| "val_loss": 0.1829780256617331, |
| "val_weighted_f1": 0.937134316978444, |
| "updates": 552 |
| }, |
| { |
| "epoch": 102, |
| "train_loss": 0.13150832892218958, |
| "val_war": 0.9401197604790419, |
| "val_uar": 0.8376741282212503, |
| "val_loss": 0.18057064132062262, |
| "val_weighted_f1": 0.9388598979782949, |
| "updates": 552 |
| }, |
| { |
| "epoch": 103, |
| "train_loss": 0.13130041319184868, |
| "val_war": 0.9416167664670658, |
| "val_uar": 0.8326645222829119, |
| "val_loss": 0.17874055357274418, |
| "val_weighted_f1": 0.9401820757548326, |
| "updates": 552 |
| }, |
| { |
| "epoch": 104, |
| "train_loss": 0.1309262527849686, |
| "val_war": 0.9391217564870259, |
| "val_uar": 0.8192555209490798, |
| "val_loss": 0.1813111626459453, |
| "val_weighted_f1": 0.9374083180236855, |
| "updates": 551 |
| }, |
| { |
| "epoch": 105, |
| "train_loss": 0.13059712380540586, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8355117432097643, |
| "val_loss": 0.18196073498673543, |
| "val_weighted_f1": 0.9384874870992324, |
| "updates": 552 |
| }, |
| { |
| "epoch": 106, |
| "train_loss": 0.1309271764687037, |
| "val_war": 0.939620758483034, |
| "val_uar": 0.8309386119751484, |
| "val_loss": 0.1815571985439864, |
| "val_weighted_f1": 0.9382253811515258, |
| "updates": 552 |
| }, |
| { |
| "epoch": 107, |
| "train_loss": 0.13030955485033385, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8375510215242851, |
| "val_loss": 0.18260493749629952, |
| "val_weighted_f1": 0.9364580918966305, |
| "updates": 552 |
| }, |
| { |
| "epoch": 108, |
| "train_loss": 0.13109447989802978, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8295682435909755, |
| "val_loss": 0.18228361816582328, |
| "val_weighted_f1": 0.9371732495935052, |
| "updates": 552 |
| }, |
| { |
| "epoch": 109, |
| "train_loss": 0.1309809476484554, |
| "val_war": 0.93812375249501, |
| "val_uar": 0.8295546790308276, |
| "val_loss": 0.1809846424979126, |
| "val_weighted_f1": 0.936760537702893, |
| "updates": 551 |
| }, |
| { |
| "epoch": 110, |
| "train_loss": 0.13025828525673044, |
| "val_war": 0.9366267465069861, |
| "val_uar": 0.8212696807333403, |
| "val_loss": 0.18340231420215733, |
| "val_weighted_f1": 0.9350511961723915, |
| "updates": 552 |
| }, |
| { |
| "epoch": 111, |
| "train_loss": 0.1309905729330597, |
| "val_war": 0.9391217564870259, |
| "val_uar": 0.8305370055494455, |
| "val_loss": 0.18326309073471023, |
| "val_weighted_f1": 0.9377237998088513, |
| "updates": 552 |
| }, |
| { |
| "epoch": 112, |
| "train_loss": 0.13059631766364443, |
| "val_war": 0.9401197604790419, |
| "val_uar": 0.8289526552200405, |
| "val_loss": 0.18176157885563826, |
| "val_weighted_f1": 0.9387643757048428, |
| "updates": 552 |
| }, |
| { |
| "epoch": 113, |
| "train_loss": 0.13075721938789658, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8224311209191706, |
| "val_loss": 0.1838092332977259, |
| "val_weighted_f1": 0.936078411773883, |
| "updates": 552 |
| }, |
| { |
| "epoch": 114, |
| "train_loss": 0.13033562467581936, |
| "val_war": 0.9391217564870259, |
| "val_uar": 0.8283881850981074, |
| "val_loss": 0.1832540118825174, |
| "val_weighted_f1": 0.9377460738926298, |
| "updates": 552 |
| }, |
| { |
| "epoch": 115, |
| "train_loss": 0.13043150780743076, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8254178574742898, |
| "val_loss": 0.18307361018633891, |
| "val_weighted_f1": 0.9371398367088126, |
| "updates": 552 |
| }, |
| { |
| "epoch": 116, |
| "train_loss": 0.1303248384266857, |
| "val_war": 0.93812375249501, |
| "val_uar": 0.8275117448642374, |
| "val_loss": 0.18226855175342865, |
| "val_weighted_f1": 0.9367804571542246, |
| "updates": 551 |
| }, |
| { |
| "epoch": 117, |
| "train_loss": 0.13041031554794658, |
| "val_war": 0.9391217564870259, |
| "val_uar": 0.8346488675360422, |
| "val_loss": 0.18322986739124367, |
| "val_weighted_f1": 0.9378908809420781, |
| "updates": 552 |
| }, |
| { |
| "epoch": 118, |
| "train_loss": 0.1301715045130683, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8254178574742898, |
| "val_loss": 0.1822983456883364, |
| "val_weighted_f1": 0.9371398367088126, |
| "updates": 552 |
| }, |
| { |
| "epoch": 119, |
| "train_loss": 0.13018532919854012, |
| "val_war": 0.938622754491018, |
| "val_uar": 0.8274954948932551, |
| "val_loss": 0.18285892469559362, |
| "val_weighted_f1": 0.9372411415559952, |
| "updates": 552 |
| }, |
| { |
| "epoch": 120, |
| "train_loss": 0.12974648134948905, |
| "val_war": 0.937624750499002, |
| "val_uar": 0.8244355309556717, |
| "val_loss": 0.1837939127179201, |
| "val_weighted_f1": 0.9361600602708166, |
| "updates": 552 |
| } |
| ], |
| "test": { |
| "n": 2047, |
| "labels_scored": [ |
| 0, |
| 1, |
| 2, |
| 3, |
| 4, |
| 5 |
| ], |
| "accuracy": 0.93209574987787, |
| "uar": 0.8140172329205021, |
| "macro_f1": 0.8289812043522109, |
| "weighted_f1": 0.931335012523836, |
| "per_class": { |
| "angry": { |
| "precision": 0.8936170212765957, |
| "recall": 0.9230769230769231, |
| "f1": 0.9081081081081082, |
| "support": 273 |
| }, |
| "disgust": { |
| "precision": 0.6666666666666666, |
| "recall": 0.4444444444444444, |
| "f1": 0.5333333333333333, |
| "support": 18 |
| }, |
| "fear": { |
| "precision": 0.7307692307692307, |
| "recall": 0.6867469879518072, |
| "f1": 0.7080745341614907, |
| "support": 83 |
| }, |
| "happy": { |
| "precision": 0.9740990990990991, |
| "recall": 0.9686450167973124, |
| "f1": 0.9713644020213363, |
| "support": 893 |
| }, |
| "sad": { |
| "precision": 0.9240837696335078, |
| "recall": 0.9192708333333334, |
| "f1": 0.9216710182767625, |
| "support": 384 |
| }, |
| "surprise": { |
| "precision": 0.9209876543209876, |
| "recall": 0.9419191919191919, |
| "f1": 0.9313358302122348, |
| "support": 396 |
| } |
| }, |
| "confusion_matrix": [ |
| [ |
| 252, |
| 1, |
| 3, |
| 8, |
| 6, |
| 3 |
| ], |
| [ |
| 6, |
| 8, |
| 0, |
| 1, |
| 1, |
| 2 |
| ], |
| [ |
| 2, |
| 0, |
| 57, |
| 0, |
| 9, |
| 15 |
| ], |
| [ |
| 8, |
| 0, |
| 0, |
| 865, |
| 10, |
| 10 |
| ], |
| [ |
| 12, |
| 2, |
| 8, |
| 7, |
| 353, |
| 2 |
| ], |
| [ |
| 2, |
| 1, |
| 10, |
| 7, |
| 3, |
| 373 |
| ] |
| ], |
| "report": " precision recall f1-score support\n\n angry 0.89 0.92 0.91 273\n disgust 0.67 0.44 0.53 18\n fear 0.73 0.69 0.71 83\n happy 0.97 0.97 0.97 893\n sad 0.92 0.92 0.92 384\n surprise 0.92 0.94 0.93 396\n\n accuracy 0.93 2047\n macro avg 0.85 0.81 0.83 2047\nweighted avg 0.93 0.93 0.93 2047\n" |
| }, |
| "test_evaluation_time_sec": 14.172675371170044, |
| "test_evaluation_peak_mem_gb": 0.562807559967041 |
| } |
