training_stats/phase2_metrics.json

{
  "phase": "phase2",
  "epochs": [
    1,
    2,
    3,
    4,
    5,
    6,
    7,
    8,
    9,
    10,
    11,
    12,
    13,
    14,
    15,
    16,
    17,
    18,
    19,
    20,
    21,
    22,
    23,
    24,
    25,
    26,
    27,
    28,
    29,
    30,
    31,
    32,
    33,
    34,
    35,
    36,
    37,
    38,
    39,
    40,
    41,
    42,
    43,
    44,
    45,
    46,
    47,
    48,
    49,
    50
  ],
  "train_loss": [
    0.24354856795753027,
    0.14387506532175465,
    0.1281819286324175,
    0.11973353732743026,
    0.11428011300439184,
    0.11022612493640627,
    0.10729721669864273,
    0.10502310001456287,
    0.10317333833280277,
    0.10161897259229963,
    0.10035178800307301,
    0.09909806067466945,
    0.09810729952710079,
    0.09719687691523735,
    0.09632332703131692,
    0.0955977276378934,
    0.09502739801094617,
    0.094355367772726,
    0.09382829038543074,
    0.09324587700372483,
    0.09266752111831368,
    0.09225670972109853,
    0.09185501903597039,
    0.09141435898987466,
    0.09111432309392509,
    0.09071751090677238,
    0.09033080931104275,
    0.09007805984168538,
    0.08986947253204988,
    0.08952355235740599,
    0.08934615796528218,
    0.08897776612111577,
    0.08882573709941613,
    0.08862475911166975,
    0.08850430102068246,
    0.0882452945889174,
    0.08803215349763595,
    0.08790178284841926,
    0.08792450884275665,
    0.08777137948309455,
    0.08768456586053335,
    0.0875368634736258,
    0.08746657900878885,
    0.08733860833434025,
    0.08737059591918282,
    0.087294191543628,
    0.08716358275036648,
    0.08720421640594661,
    0.08716494832471731,
    0.08717396302563293
  ],
  "val_loss": [
    0.12154528312198631,
    0.09832304232829325,
    0.08902931560177703,
    0.08291382267996594,
    0.07845750635674408,
    0.07589214435359976,
    0.07343193763742098,
    0.07170625012349319,
    0.07002376107213733,
    0.06894454053482851,
    0.06793339362552574,
    0.06696961053439208,
    0.06628897291299578,
    0.06615510113440357,
    0.06493215745797863,
    0.0643177818473395,
    0.06415080185140576,
    0.06378142737426439,
    0.06311914445207761,
    0.0630511126226491,
    0.06248988477406917,
    0.06210367412523584,
    0.06193486649218622,
    0.061674018816517163,
    0.06147365603927855,
    0.06100164517551452,
    0.06107486015534377,
    0.0608744491283242,
    0.060593874202118266,
    0.06045886799618527,
    0.060196686103507445,
    0.0600850325044211,
    0.06000507507357631,
    0.06000771318693777,
    0.05984950957967116,
    0.059681328511363395,
    0.05965545929752074,
    0.059552186262887996,
    0.05945843880673548,
    0.05929617923092436,
    0.05935625158048965,
    0.05923473901367999,
    0.05928887761719115,
    0.059216197945542996,
    0.059152334331869484,
    0.05911746904164463,
    0.059118302164224534,
    0.059132881529696354,
    0.059078274975280023,
    0.059064933994749526
  ],
  "learning_rate": [
    0.0001,
    9.990232305719945e-05,
    9.960967771506666e-05,
    9.912321891107009e-05,
    9.844486647586723e-05,
    9.757729755661011e-05,
    9.652393605146844e-05,
    9.528893909706797e-05,
    9.387718066217124e-05,
    9.229423231234974e-05,
    9.054634122155989e-05,
    8.864040551740155e-05,
    8.658394705735986e-05,
    8.438508174347008e-05,
    8.205248749256013e-05,
    7.959536998847742e-05,
    7.702342635146032e-05,
    7.434680686803489e-05,
    7.157607493247108e-05,
    6.872216535789155e-05,
    6.579634122155988e-05,
    6.28101494146603e-05,
    5.9775375071993364e-05,
    5.670399506143306e-05,
    5.3608130716701005e-05,
    5.049999999999999e-05,
    4.7391869283298976e-05,
    4.429600493856693e-05,
    4.122462492800662e-05,
    3.8189850585339674e-05,
    3.5203658778440106e-05,
    3.227783464210844e-05,
    2.942392506752889e-05,
    2.6653193131965075e-05,
    2.3976573648539646e-05,
    2.1404630011522576e-05,
    1.8947512507439848e-05,
    1.66149182565299e-05,
    1.4416052942640132e-05,
    1.235959448259843e-05,
    1.04536587784401e-05,
    8.70576768765026e-06,
    7.122819337828748e-06,
    5.711060902932039e-06,
    4.476063948531556e-06,
    3.4227024433898985e-06,
    2.555133524132766e-06,
    1.8767810889299075e-06,
    1.3903222849333503e-06,
    1.0976769428005576e-06
  ],
  "best_epoch": 50,
  "best_val_loss": 0.059064933994749526,
  "cosine_sim": [
    0.8803246604191052,
    0.9031896384151371,
    0.9123403851215068,
    0.9183617949008465,
    0.9227495524618361,
    0.9252754476693299,
    0.9276978032844322,
    0.9293969420938043,
    0.9310535484248096,
    0.9321161643998163,
    0.9331117559958984,
    0.9340607108057918,
    0.934730878165057,
    0.9348626897499726,
    0.9360668202778241,
    0.9366717426268546,
    0.9368361541936109,
    0.9371998441350591,
    0.9378519391751027,
    0.9379189245932333,
    0.9384715200902464,
    0.9388517861490374,
    0.9390179956400836,
    0.9392748343813289,
    0.9394721114361012,
    0.939936862574206,
    0.9398647752610055,
    0.9400621020161473,
    0.9403383599387275,
    0.9404712899311168,
    0.940729436872003,
    0.940839373910272,
    0.940918100190473,
    0.940915504256049,
    0.9410712719560266,
    0.9412368669046892,
    0.9412623369419301,
    0.9413640225255812,
    0.9414563282832011,
    0.9416160896137074,
    0.9415569422958611,
    0.941676585165946,
    0.9416232819075102,
    0.9416948422058686,
    0.9417577226598699,
    0.9417920493268155,
    0.9417912312694736,
    0.9417768769316726,
    0.9418306413952175,
    0.9418437798340638
  ],
  "mse_loss": [
    0.0018699278392399232,
    0.0015126625925453532,
    0.0013696821880768294,
    0.00127559767670564,
    0.001207038966155293,
    0.0011675718637347385,
    0.0011297225261347609,
    0.0011031734824713854,
    0.0010772890438585801,
    0.0010606856451270072,
    0.0010451295368678242,
    0.0010303021101556063,
    0.00101983076171461,
    0.0010177711945132458,
    0.0009989566824588183,
    0.000989504741971848,
    0.0009869358149120348,
    0.0009812531357461416,
    0.0009710641658946022,
    0.000970017522873191,
    0.0009613832517935371,
    0.0009554415463224739,
    0.0009528445080505335,
    0.0009488314653544485,
    0.0009457489602744124,
    0.0009384872600994631,
    0.0009396136352639493,
    0.0009365303923624817,
    0.0009322138548058619,
    0.0009301368363767497,
    0.0009261032697992647,
    0.0009243855283393323,
    0.0009231554125566166,
    0.0009231959929101513,
    0.0009207620990585659,
    0.0009181746904546202,
    0.0009177767016036479,
    0.0009161878885977246,
    0.0009147456191756108,
    0.0009122493182629667,
    0.0009131735089716051,
    0.0009113040870289553,
    0.0009121369841197194,
    0.0009110188403575158,
    0.0009100363171823487,
    0.0009094999358011363,
    0.0009095127483619166,
    0.000909737044970064,
    0.0009088969434139779,
    0.0009086916967340791
  ],
  "cosine_loss": [
    0.11967535527260811,
    0.09681037974518698,
    0.08765963337442896,
    0.08163822495991045,
    0.07725046745112708,
    0.07472457263518024,
    0.07230221518540048,
    0.07060307667807893,
    0.06894647199530024,
    0.06788385487921364,
    0.06688826408114161,
    0.06593930850307027,
    0.06526914207233084,
    0.06513732988331411,
    0.06393320086571547,
    0.06332827718169601,
    0.06316386605496402,
    0.06280017425646414,
    0.06214808030529423,
    0.06208109508853059,
    0.06152850158654295,
    0.06114823249605922,
    0.06098202199071974,
    0.060725187351395774,
    0.06052790705983464,
    0.06006315796896144,
    0.06013524652765439,
    0.05993791879923852,
    0.059661660276286235,
    0.05952873112292619,
    0.059270582783657745,
    0.059160647024442486,
    0.059081919752322395,
    0.059084517163438,
    0.05892874748708011,
    0.058763153880864415,
    0.05873768261304727,
    0.058635998394217216,
    0.058543693218324276,
    0.05838392999347624,
    0.05844307807950048,
    0.058323434978216376,
    0.058376740694075856,
    0.058305179075644656,
    0.0582422979280457,
    0.05820796918776539,
    0.05820878931098395,
    0.05822314441696302,
    0.05816937805534722,
    0.05815624228274977
  ]
}