| { |
| "config": { |
| "name": "esmc_300M", |
| "model_name": "Synthyra/ESMplusplus_small", |
| "pooling": "mean", |
| "space_residues": false, |
| "max_len": 2048, |
| "use_t5_encoder": false, |
| "use_automaskedlm": true, |
| "trust_remote_code": true |
| }, |
| "result": { |
| "encoder": "esmc_300M", |
| "run_dir": "esmc_300M_sym_v2", |
| "run_tag": "sym_v2", |
| "model_name": "Synthyra/ESMplusplus_small", |
| "hidden_dim": 960, |
| "max_len": 2048, |
| "pooling": "mean", |
| "head": "mlp", |
| "moe_experts": null, |
| "moe_top_k": null, |
| "best_val_rmse": 1.343891979080109, |
| "val_rmse": 1.343891979080109, |
| "val_mae": 0.874987006187439, |
| "val_spearman": 0.8618226129144756, |
| "val_n": 666, |
| "s90_mae": 1.5350065231323242, |
| "s90_rmse": 2.0848605343945614, |
| "s90_pearson": 0.6011604733222113, |
| "s90_spearman": 0.5510109375406975, |
| "s90_n": 90, |
| "s79_mae": 1.5322797298431396, |
| "s79_rmse": 2.0387059627152926, |
| "s79_pearson": 0.698892595570155, |
| "s79_spearman": 0.7051683531133182, |
| "s79_n_complexes": 79, |
| "s79_n_pairs": 79, |
| "status": "ok" |
| }, |
| "history": [ |
| { |
| "epoch": 1, |
| "train_loss": 5.2500034169053755, |
| "rmse": 2.3560124366056368, |
| "mae": 1.936526894569397, |
| "spearman": 0.7146812219244423, |
| "n": 666 |
| }, |
| { |
| "epoch": 2, |
| "train_loss": 3.16614606161958, |
| "rmse": 1.8153762358070458, |
| "mae": 1.3361821174621582, |
| "spearman": 0.7877017189484826, |
| "n": 666 |
| }, |
| { |
| "epoch": 3, |
| "train_loss": 2.7385075062827213, |
| "rmse": 1.770573391254816, |
| "mae": 1.3470638990402222, |
| "spearman": 0.7908117287889782, |
| "n": 666 |
| }, |
| { |
| "epoch": 4, |
| "train_loss": 2.5775398382301797, |
| "rmse": 1.9354384745124416, |
| "mae": 1.4594480991363525, |
| "spearman": 0.7951679210899366, |
| "n": 666 |
| }, |
| { |
| "epoch": 5, |
| "train_loss": 2.296075133634992, |
| "rmse": 1.7771083130893897, |
| "mae": 1.3618661165237427, |
| "spearman": 0.7878699347478659, |
| "n": 666 |
| }, |
| { |
| "epoch": 6, |
| "train_loss": 2.1632901883809876, |
| "rmse": 1.571707641914237, |
| "mae": 1.0820832252502441, |
| "spearman": 0.8242117697646001, |
| "n": 666 |
| }, |
| { |
| "epoch": 7, |
| "train_loss": 2.0487842086902845, |
| "rmse": 1.7326433627220126, |
| "mae": 1.2997156381607056, |
| "spearman": 0.8067283923820779, |
| "n": 666 |
| }, |
| { |
| "epoch": 8, |
| "train_loss": 1.9539089389904762, |
| "rmse": 1.5323771007102422, |
| "mae": 1.0768132209777832, |
| "spearman": 0.8331224830364253, |
| "n": 666 |
| }, |
| { |
| "epoch": 9, |
| "train_loss": 1.884503437786536, |
| "rmse": 1.4836944626245834, |
| "mae": 1.0211477279663086, |
| "spearman": 0.8453298885675242, |
| "n": 666 |
| }, |
| { |
| "epoch": 10, |
| "train_loss": 1.7379148220976754, |
| "rmse": 1.5289483190800148, |
| "mae": 1.0419483184814453, |
| "spearman": 0.8279244517802901, |
| "n": 666 |
| }, |
| { |
| "epoch": 11, |
| "train_loss": 1.637459601849805, |
| "rmse": 1.50807899136166, |
| "mae": 1.0249881744384766, |
| "spearman": 0.8349588743741613, |
| "n": 666 |
| }, |
| { |
| "epoch": 12, |
| "train_loss": 1.570613271213623, |
| "rmse": 1.5877679493567813, |
| "mae": 1.047106385231018, |
| "spearman": 0.8173064320570957, |
| "n": 666 |
| }, |
| { |
| "epoch": 13, |
| "train_loss": 1.5191953086877446, |
| "rmse": 1.523368716521125, |
| "mae": 1.0183935165405273, |
| "spearman": 0.835533415298498, |
| "n": 666 |
| }, |
| { |
| "epoch": 14, |
| "train_loss": 1.4484552232677594, |
| "rmse": 1.4635659302932218, |
| "mae": 0.9944096207618713, |
| "spearman": 0.845873219852843, |
| "n": 666 |
| }, |
| { |
| "epoch": 15, |
| "train_loss": 1.3779972628416162, |
| "rmse": 1.4513941475237058, |
| "mae": 0.9942841529846191, |
| "spearman": 0.8415644130046825, |
| "n": 666 |
| }, |
| { |
| "epoch": 16, |
| "train_loss": 1.3456517325965984, |
| "rmse": 1.4878169259075829, |
| "mae": 1.0041481256484985, |
| "spearman": 0.8394071857523199, |
| "n": 666 |
| }, |
| { |
| "epoch": 17, |
| "train_loss": 1.283067804568361, |
| "rmse": 1.5455195335503902, |
| "mae": 1.0542510747909546, |
| "spearman": 0.8402040797740962, |
| "n": 666 |
| }, |
| { |
| "epoch": 18, |
| "train_loss": 1.2671775079257779, |
| "rmse": 1.4594022466497183, |
| "mae": 0.9660055041313171, |
| "spearman": 0.8463883766343031, |
| "n": 666 |
| }, |
| { |
| "epoch": 19, |
| "train_loss": 1.1808337405414078, |
| "rmse": 1.463115354014868, |
| "mae": 0.9558651447296143, |
| "spearman": 0.8468865940422482, |
| "n": 666 |
| }, |
| { |
| "epoch": 20, |
| "train_loss": 1.1422315512511436, |
| "rmse": 1.4254628283130863, |
| "mae": 0.9524824023246765, |
| "spearman": 0.8563309012442194, |
| "n": 666 |
| }, |
| { |
| "epoch": 21, |
| "train_loss": 1.082284438918762, |
| "rmse": 1.4807769535103399, |
| "mae": 0.9906092286109924, |
| "spearman": 0.8476697202161565, |
| "n": 666 |
| }, |
| { |
| "epoch": 22, |
| "train_loss": 1.0354805702359686, |
| "rmse": 1.528388015173751, |
| "mae": 1.04389226436615, |
| "spearman": 0.8509532998653907, |
| "n": 666 |
| }, |
| { |
| "epoch": 23, |
| "train_loss": 1.0098871153456515, |
| "rmse": 1.3559876693654447, |
| "mae": 0.9069312810897827, |
| "spearman": 0.8593247965836093, |
| "n": 666 |
| }, |
| { |
| "epoch": 24, |
| "train_loss": 0.9503703362235882, |
| "rmse": 1.343891979080109, |
| "mae": 0.874987006187439, |
| "spearman": 0.8618226129144756, |
| "n": 666 |
| }, |
| { |
| "epoch": 25, |
| "train_loss": 0.925024896347635, |
| "rmse": 1.4017211892269734, |
| "mae": 0.9141939282417297, |
| "spearman": 0.8530537717668791, |
| "n": 666 |
| }, |
| { |
| "epoch": 26, |
| "train_loss": 0.8724437099130326, |
| "rmse": 1.390283907088061, |
| "mae": 0.9096393585205078, |
| "spearman": 0.8577643334770891, |
| "n": 666 |
| }, |
| { |
| "epoch": 27, |
| "train_loss": 0.8766851046059907, |
| "rmse": 1.363666425359147, |
| "mae": 0.8954227566719055, |
| "spearman": 0.8578916719195002, |
| "n": 666 |
| }, |
| { |
| "epoch": 28, |
| "train_loss": 0.8230470686901363, |
| "rmse": 1.385749370618165, |
| "mae": 0.9481036067008972, |
| "spearman": 0.8584835841364484, |
| "n": 666 |
| }, |
| { |
| "epoch": 29, |
| "train_loss": 0.8009588031335821, |
| "rmse": 1.4175805435307196, |
| "mae": 0.937690794467926, |
| "spearman": 0.8479630893759342, |
| "n": 666 |
| }, |
| { |
| "epoch": 30, |
| "train_loss": 0.7837735650966295, |
| "rmse": 1.3701328486052542, |
| "mae": 0.8667737245559692, |
| "spearman": 0.8634084967125738, |
| "n": 666 |
| }, |
| { |
| "epoch": 31, |
| "train_loss": 0.7858880045990005, |
| "rmse": 1.3988896070666528, |
| "mae": 0.9008198976516724, |
| "spearman": 0.8655306318431181, |
| "n": 666 |
| } |
| ] |
| } |
