Instructions to use circlemachinelearning/bart-email-multitask with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use circlemachinelearning/bart-email-multitask with Transformers:
# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("circlemachinelearning/bart-email-multitask") model = AutoModelForSeq2SeqLM.from_pretrained("circlemachinelearning/bart-email-multitask", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Invalid JSON:Unexpected token 'I', ..."ad_norm": Infinity,
"... is not valid JSON
| { | |
| "best_global_step": 3000, | |
| "best_metric": 51.14, | |
| "best_model_checkpoint": "./bart-email-multitask-updated/checkpoint-3000", | |
| "epoch": 1.3807715543281263, | |
| "eval_steps": 500, | |
| "global_step": 4000, | |
| "is_hyper_param_search": false, | |
| "is_local_process_zero": true, | |
| "is_world_process_zero": true, | |
| "log_history": [ | |
| { | |
| "epoch": 0.0172607232243031, | |
| "grad_norm": 3.8177006244659424, | |
| "learning_rate": 1.2250000000000001e-06, | |
| "loss": 0.4507, | |
| "step": 50 | |
| }, | |
| { | |
| "epoch": 0.0345214464486062, | |
| "grad_norm": Infinity, | |
| "learning_rate": 2.475e-06, | |
| "loss": 0.4331, | |
| "step": 100 | |
| }, | |
| { | |
| "epoch": 0.051782169672909294, | |
| "grad_norm": 2.9417564868927, | |
| "learning_rate": 3.7250000000000003e-06, | |
| "loss": 0.4042, | |
| "step": 150 | |
| }, | |
| { | |
| "epoch": 0.0690428928972124, | |
| "grad_norm": 2.8877062797546387, | |
| "learning_rate": 4.975000000000001e-06, | |
| "loss": 0.397, | |
| "step": 200 | |
| }, | |
| { | |
| "epoch": 0.08630361612151549, | |
| "grad_norm": 3.343304395675659, | |
| "learning_rate": 4.956203074722918e-06, | |
| "loss": 0.3892, | |
| "step": 250 | |
| }, | |
| { | |
| "epoch": 0.10356433934581859, | |
| "grad_norm": 3.0275661945343018, | |
| "learning_rate": 4.911512334644262e-06, | |
| "loss": 0.3891, | |
| "step": 300 | |
| }, | |
| { | |
| "epoch": 0.12082506257012168, | |
| "grad_norm": 3.045884847640991, | |
| "learning_rate": 4.866821594565607e-06, | |
| "loss": 0.3656, | |
| "step": 350 | |
| }, | |
| { | |
| "epoch": 0.1380857857944248, | |
| "grad_norm": 2.7904183864593506, | |
| "learning_rate": 4.82213085448695e-06, | |
| "loss": 0.365, | |
| "step": 400 | |
| }, | |
| { | |
| "epoch": 0.15534650901872787, | |
| "grad_norm": 3.6787497997283936, | |
| "learning_rate": 4.777440114408295e-06, | |
| "loss": 0.3594, | |
| "step": 450 | |
| }, | |
| { | |
| "epoch": 0.17260723224303098, | |
| "grad_norm": 3.1564106941223145, | |
| "learning_rate": 4.732749374329639e-06, | |
| "loss": 0.3719, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.17260723224303098, | |
| "eval_gen_len": 55.68621291448517, | |
| "eval_loss": 0.5641192197799683, | |
| "eval_rouge1": 49.82, | |
| "eval_rouge2": 35.56, | |
| "eval_rougeL": 43.37, | |
| "eval_rougeLsum": 43.9, | |
| "eval_runtime": 5387.3225, | |
| "eval_samples_per_second": 1.064, | |
| "eval_steps_per_second": 0.266, | |
| "step": 500 | |
| }, | |
| { | |
| "epoch": 0.1898679554673341, | |
| "grad_norm": 3.0305216312408447, | |
| "learning_rate": 4.688058634250984e-06, | |
| "loss": 0.3849, | |
| "step": 550 | |
| }, | |
| { | |
| "epoch": 0.20712867869163717, | |
| "grad_norm": 3.079951524734497, | |
| "learning_rate": 4.643367894172328e-06, | |
| "loss": 0.3802, | |
| "step": 600 | |
| }, | |
| { | |
| "epoch": 0.22438940191594028, | |
| "grad_norm": 2.7168169021606445, | |
| "learning_rate": 4.598677154093672e-06, | |
| "loss": 0.3887, | |
| "step": 650 | |
| }, | |
| { | |
| "epoch": 0.24165012514024337, | |
| "grad_norm": 2.76423716545105, | |
| "learning_rate": 4.553986414015016e-06, | |
| "loss": 0.3825, | |
| "step": 700 | |
| }, | |
| { | |
| "epoch": 0.2589108483645465, | |
| "grad_norm": 4.16721248626709, | |
| "learning_rate": 4.509295673936361e-06, | |
| "loss": 0.3819, | |
| "step": 750 | |
| }, | |
| { | |
| "epoch": 0.2761715715888496, | |
| "grad_norm": 2.646681785583496, | |
| "learning_rate": 4.464604933857705e-06, | |
| "loss": 0.3813, | |
| "step": 800 | |
| }, | |
| { | |
| "epoch": 0.2934322948131527, | |
| "grad_norm": 3.4927141666412354, | |
| "learning_rate": 4.419914193779049e-06, | |
| "loss": 0.3793, | |
| "step": 850 | |
| }, | |
| { | |
| "epoch": 0.31069301803745575, | |
| "grad_norm": 2.364363193511963, | |
| "learning_rate": 4.375223453700393e-06, | |
| "loss": 0.3815, | |
| "step": 900 | |
| }, | |
| { | |
| "epoch": 0.32795374126175886, | |
| "grad_norm": 2.7475292682647705, | |
| "learning_rate": 4.330532713621738e-06, | |
| "loss": 0.3652, | |
| "step": 950 | |
| }, | |
| { | |
| "epoch": 0.34521446448606197, | |
| "grad_norm": 3.994381904602051, | |
| "learning_rate": 4.285841973543082e-06, | |
| "loss": 0.39, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.34521446448606197, | |
| "eval_gen_len": 53.06160558464224, | |
| "eval_loss": 0.5498749613761902, | |
| "eval_rouge1": 50.66, | |
| "eval_rouge2": 36.17, | |
| "eval_rougeL": 44.07, | |
| "eval_rougeLsum": 44.56, | |
| "eval_runtime": 5246.3152, | |
| "eval_samples_per_second": 1.092, | |
| "eval_steps_per_second": 0.273, | |
| "step": 1000 | |
| }, | |
| { | |
| "epoch": 0.3624751877103651, | |
| "grad_norm": 4.5150227546691895, | |
| "learning_rate": 4.241151233464427e-06, | |
| "loss": 0.4227, | |
| "step": 1050 | |
| }, | |
| { | |
| "epoch": 0.3797359109346682, | |
| "grad_norm": 2.6075778007507324, | |
| "learning_rate": 4.19646049338577e-06, | |
| "loss": 0.4255, | |
| "step": 1100 | |
| }, | |
| { | |
| "epoch": 0.39699663415897124, | |
| "grad_norm": 3.93216609954834, | |
| "learning_rate": 4.151769753307115e-06, | |
| "loss": 0.4192, | |
| "step": 1150 | |
| }, | |
| { | |
| "epoch": 0.41425735738327435, | |
| "grad_norm": 3.318143606185913, | |
| "learning_rate": 4.107079013228459e-06, | |
| "loss": 0.3984, | |
| "step": 1200 | |
| }, | |
| { | |
| "epoch": 0.43151808060757746, | |
| "grad_norm": 3.0835282802581787, | |
| "learning_rate": 4.062388273149804e-06, | |
| "loss": 0.4086, | |
| "step": 1250 | |
| }, | |
| { | |
| "epoch": 0.44877880383188057, | |
| "grad_norm": 2.782550811767578, | |
| "learning_rate": 4.017697533071148e-06, | |
| "loss": 0.4114, | |
| "step": 1300 | |
| }, | |
| { | |
| "epoch": 0.4660395270561837, | |
| "grad_norm": 3.2588839530944824, | |
| "learning_rate": 3.973006792992492e-06, | |
| "loss": 0.4304, | |
| "step": 1350 | |
| }, | |
| { | |
| "epoch": 0.48330025028048673, | |
| "grad_norm": 3.4788544178009033, | |
| "learning_rate": 3.928316052913836e-06, | |
| "loss": 0.4047, | |
| "step": 1400 | |
| }, | |
| { | |
| "epoch": 0.5005609735047899, | |
| "grad_norm": 3.0864784717559814, | |
| "learning_rate": 3.883625312835181e-06, | |
| "loss": 0.4185, | |
| "step": 1450 | |
| }, | |
| { | |
| "epoch": 0.517821696729093, | |
| "grad_norm": 2.8312442302703857, | |
| "learning_rate": 3.838934572756525e-06, | |
| "loss": 0.4325, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.517821696729093, | |
| "eval_gen_len": 54.41937172774869, | |
| "eval_loss": 0.5411620736122131, | |
| "eval_rouge1": 50.59, | |
| "eval_rouge2": 36.52, | |
| "eval_rougeL": 44.19, | |
| "eval_rougeLsum": 44.63, | |
| "eval_runtime": 5226.6165, | |
| "eval_samples_per_second": 1.096, | |
| "eval_steps_per_second": 0.274, | |
| "step": 1500 | |
| }, | |
| { | |
| "epoch": 0.535082419953396, | |
| "grad_norm": 3.31876540184021, | |
| "learning_rate": 3.7942438326778697e-06, | |
| "loss": 0.4191, | |
| "step": 1550 | |
| }, | |
| { | |
| "epoch": 0.5523431431776992, | |
| "grad_norm": 3.1453733444213867, | |
| "learning_rate": 3.7495530925992137e-06, | |
| "loss": 0.4591, | |
| "step": 1600 | |
| }, | |
| { | |
| "epoch": 0.5696038664020022, | |
| "grad_norm": 3.1156108379364014, | |
| "learning_rate": 3.7048623525205578e-06, | |
| "loss": 0.4243, | |
| "step": 1650 | |
| }, | |
| { | |
| "epoch": 0.5868645896263054, | |
| "grad_norm": 3.0627033710479736, | |
| "learning_rate": 3.6601716124419022e-06, | |
| "loss": 0.4518, | |
| "step": 1700 | |
| }, | |
| { | |
| "epoch": 0.6041253128506084, | |
| "grad_norm": 3.0280168056488037, | |
| "learning_rate": 3.6154808723632467e-06, | |
| "loss": 0.4467, | |
| "step": 1750 | |
| }, | |
| { | |
| "epoch": 0.6213860360749115, | |
| "grad_norm": 3.178391456604004, | |
| "learning_rate": 3.570790132284591e-06, | |
| "loss": 0.4411, | |
| "step": 1800 | |
| }, | |
| { | |
| "epoch": 0.6386467592992147, | |
| "grad_norm": 3.445444345474243, | |
| "learning_rate": 3.526099392205935e-06, | |
| "loss": 0.425, | |
| "step": 1850 | |
| }, | |
| { | |
| "epoch": 0.6559074825235177, | |
| "grad_norm": 2.4380524158477783, | |
| "learning_rate": 3.4814086521272793e-06, | |
| "loss": 0.4237, | |
| "step": 1900 | |
| }, | |
| { | |
| "epoch": 0.6731682057478209, | |
| "grad_norm": 3.591120958328247, | |
| "learning_rate": 3.4367179120486237e-06, | |
| "loss": 0.4465, | |
| "step": 1950 | |
| }, | |
| { | |
| "epoch": 0.6904289289721239, | |
| "grad_norm": 2.699303388595581, | |
| "learning_rate": 3.3920271719699682e-06, | |
| "loss": 0.4328, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.6904289289721239, | |
| "eval_gen_len": 54.16980802792321, | |
| "eval_loss": 0.5556456446647644, | |
| "eval_rouge1": 50.8, | |
| "eval_rouge2": 36.6, | |
| "eval_rougeL": 44.19, | |
| "eval_rougeLsum": 44.7, | |
| "eval_runtime": 5233.7113, | |
| "eval_samples_per_second": 1.095, | |
| "eval_steps_per_second": 0.274, | |
| "step": 2000 | |
| }, | |
| { | |
| "epoch": 0.707689652196427, | |
| "grad_norm": 5.292072772979736, | |
| "learning_rate": 3.3473364318913127e-06, | |
| "loss": 0.4216, | |
| "step": 2050 | |
| }, | |
| { | |
| "epoch": 0.7249503754207302, | |
| "grad_norm": 2.9521191120147705, | |
| "learning_rate": 3.3026456918126563e-06, | |
| "loss": 0.4332, | |
| "step": 2100 | |
| }, | |
| { | |
| "epoch": 0.7422110986450332, | |
| "grad_norm": 4.375521183013916, | |
| "learning_rate": 3.2579549517340008e-06, | |
| "loss": 0.4392, | |
| "step": 2150 | |
| }, | |
| { | |
| "epoch": 0.7594718218693364, | |
| "grad_norm": 3.5504395961761475, | |
| "learning_rate": 3.2132642116553453e-06, | |
| "loss": 0.432, | |
| "step": 2200 | |
| }, | |
| { | |
| "epoch": 0.7767325450936394, | |
| "grad_norm": 3.7240469455718994, | |
| "learning_rate": 3.1685734715766897e-06, | |
| "loss": 0.4278, | |
| "step": 2250 | |
| }, | |
| { | |
| "epoch": 0.7939932683179425, | |
| "grad_norm": 2.717268943786621, | |
| "learning_rate": 3.123882731498034e-06, | |
| "loss": 0.4145, | |
| "step": 2300 | |
| }, | |
| { | |
| "epoch": 0.8112539915422456, | |
| "grad_norm": 2.7192165851593018, | |
| "learning_rate": 3.079191991419378e-06, | |
| "loss": 0.4454, | |
| "step": 2350 | |
| }, | |
| { | |
| "epoch": 0.8285147147665487, | |
| "grad_norm": 4.111413955688477, | |
| "learning_rate": 3.0345012513407223e-06, | |
| "loss": 0.4188, | |
| "step": 2400 | |
| }, | |
| { | |
| "epoch": 0.8457754379908519, | |
| "grad_norm": 3.396127223968506, | |
| "learning_rate": 2.9898105112620668e-06, | |
| "loss": 0.4349, | |
| "step": 2450 | |
| }, | |
| { | |
| "epoch": 0.8630361612151549, | |
| "grad_norm": 2.503493309020996, | |
| "learning_rate": 2.9451197711834112e-06, | |
| "loss": 0.4133, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.8630361612151549, | |
| "eval_gen_len": 54.88080279232112, | |
| "eval_loss": 0.5358440279960632, | |
| "eval_rouge1": 50.67, | |
| "eval_rouge2": 36.36, | |
| "eval_rougeL": 44.02, | |
| "eval_rougeLsum": 44.56, | |
| "eval_runtime": 5344.428, | |
| "eval_samples_per_second": 1.072, | |
| "eval_steps_per_second": 0.268, | |
| "step": 2500 | |
| }, | |
| { | |
| "epoch": 0.880296884439458, | |
| "grad_norm": 2.877305746078491, | |
| "learning_rate": 2.9004290311047557e-06, | |
| "loss": 0.4369, | |
| "step": 2550 | |
| }, | |
| { | |
| "epoch": 0.8975576076637611, | |
| "grad_norm": 2.5334177017211914, | |
| "learning_rate": 2.8557382910260993e-06, | |
| "loss": 0.4371, | |
| "step": 2600 | |
| }, | |
| { | |
| "epoch": 0.9148183308880642, | |
| "grad_norm": 2.780935049057007, | |
| "learning_rate": 2.811047550947444e-06, | |
| "loss": 0.3965, | |
| "step": 2650 | |
| }, | |
| { | |
| "epoch": 0.9320790541123674, | |
| "grad_norm": 3.019552707672119, | |
| "learning_rate": 2.7663568108687883e-06, | |
| "loss": 0.4191, | |
| "step": 2700 | |
| }, | |
| { | |
| "epoch": 0.9493397773366704, | |
| "grad_norm": 3.7161452770233154, | |
| "learning_rate": 2.7216660707901327e-06, | |
| "loss": 0.4164, | |
| "step": 2750 | |
| }, | |
| { | |
| "epoch": 0.9666005005609735, | |
| "grad_norm": 2.9958629608154297, | |
| "learning_rate": 2.6769753307114764e-06, | |
| "loss": 0.4069, | |
| "step": 2800 | |
| }, | |
| { | |
| "epoch": 0.9838612237852766, | |
| "grad_norm": 2.9618115425109863, | |
| "learning_rate": 2.632284590632821e-06, | |
| "loss": 0.4302, | |
| "step": 2850 | |
| }, | |
| { | |
| "epoch": 1.0010356433934582, | |
| "grad_norm": 2.9072768688201904, | |
| "learning_rate": 2.5875938505541653e-06, | |
| "loss": 0.426, | |
| "step": 2900 | |
| }, | |
| { | |
| "epoch": 1.0182963666177614, | |
| "grad_norm": 3.2817041873931885, | |
| "learning_rate": 2.5429031104755098e-06, | |
| "loss": 0.3912, | |
| "step": 2950 | |
| }, | |
| { | |
| "epoch": 1.0355570898420643, | |
| "grad_norm": 2.7742252349853516, | |
| "learning_rate": 2.498212370396854e-06, | |
| "loss": 0.3576, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 1.0355570898420643, | |
| "eval_gen_len": 53.4542757417103, | |
| "eval_loss": 0.5370803475379944, | |
| "eval_rouge1": 51.14, | |
| "eval_rouge2": 36.78, | |
| "eval_rougeL": 44.59, | |
| "eval_rougeLsum": 45.09, | |
| "eval_runtime": 5216.1975, | |
| "eval_samples_per_second": 1.099, | |
| "eval_steps_per_second": 0.275, | |
| "step": 3000 | |
| }, | |
| { | |
| "epoch": 1.0528178130663675, | |
| "grad_norm": 2.5936224460601807, | |
| "learning_rate": 2.4535216303181983e-06, | |
| "loss": 0.3795, | |
| "step": 3050 | |
| }, | |
| { | |
| "epoch": 1.0700785362906706, | |
| "grad_norm": 2.727391481399536, | |
| "learning_rate": 2.4088308902395428e-06, | |
| "loss": 0.3797, | |
| "step": 3100 | |
| }, | |
| { | |
| "epoch": 1.0873392595149736, | |
| "grad_norm": 2.1693854331970215, | |
| "learning_rate": 2.364140150160887e-06, | |
| "loss": 0.3809, | |
| "step": 3150 | |
| }, | |
| { | |
| "epoch": 1.1045999827392767, | |
| "grad_norm": 3.9267098903656006, | |
| "learning_rate": 2.3194494100822313e-06, | |
| "loss": 0.3816, | |
| "step": 3200 | |
| }, | |
| { | |
| "epoch": 1.12186070596358, | |
| "grad_norm": 3.905186176300049, | |
| "learning_rate": 2.2747586700035753e-06, | |
| "loss": 0.3852, | |
| "step": 3250 | |
| }, | |
| { | |
| "epoch": 1.139121429187883, | |
| "grad_norm": 3.1914596557617188, | |
| "learning_rate": 2.23006792992492e-06, | |
| "loss": 0.3837, | |
| "step": 3300 | |
| }, | |
| { | |
| "epoch": 1.156382152412186, | |
| "grad_norm": 3.2150516510009766, | |
| "learning_rate": 2.185377189846264e-06, | |
| "loss": 0.3916, | |
| "step": 3350 | |
| }, | |
| { | |
| "epoch": 1.1736428756364892, | |
| "grad_norm": 3.4676005840301514, | |
| "learning_rate": 2.1406864497676083e-06, | |
| "loss": 0.3889, | |
| "step": 3400 | |
| }, | |
| { | |
| "epoch": 1.1909035988607923, | |
| "grad_norm": 3.015994071960449, | |
| "learning_rate": 2.095995709688953e-06, | |
| "loss": 0.3901, | |
| "step": 3450 | |
| }, | |
| { | |
| "epoch": 1.2081643220850953, | |
| "grad_norm": 3.504163980484009, | |
| "learning_rate": 2.051304969610297e-06, | |
| "loss": 0.3878, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 1.2081643220850953, | |
| "eval_gen_len": 54.42425828970332, | |
| "eval_loss": 0.5358798503875732, | |
| "eval_rouge1": 50.77, | |
| "eval_rouge2": 36.52, | |
| "eval_rougeL": 44.17, | |
| "eval_rougeLsum": 44.73, | |
| "eval_runtime": 5360.6563, | |
| "eval_samples_per_second": 1.069, | |
| "eval_steps_per_second": 0.267, | |
| "step": 3500 | |
| }, | |
| { | |
| "epoch": 1.2254250453093984, | |
| "grad_norm": 2.9530227184295654, | |
| "learning_rate": 2.0066142295316413e-06, | |
| "loss": 0.4127, | |
| "step": 3550 | |
| }, | |
| { | |
| "epoch": 1.2426857685337016, | |
| "grad_norm": 2.9822258949279785, | |
| "learning_rate": 1.9619234894529854e-06, | |
| "loss": 0.3785, | |
| "step": 3600 | |
| }, | |
| { | |
| "epoch": 1.2599464917580048, | |
| "grad_norm": 3.8967037200927734, | |
| "learning_rate": 1.91723274937433e-06, | |
| "loss": 0.4015, | |
| "step": 3650 | |
| }, | |
| { | |
| "epoch": 1.2772072149823077, | |
| "grad_norm": 2.652034044265747, | |
| "learning_rate": 1.872542009295674e-06, | |
| "loss": 0.3962, | |
| "step": 3700 | |
| }, | |
| { | |
| "epoch": 1.2944679382066109, | |
| "grad_norm": 2.3683066368103027, | |
| "learning_rate": 1.8278512692170184e-06, | |
| "loss": 0.3942, | |
| "step": 3750 | |
| }, | |
| { | |
| "epoch": 1.311728661430914, | |
| "grad_norm": 2.787900447845459, | |
| "learning_rate": 1.7831605291383628e-06, | |
| "loss": 0.3902, | |
| "step": 3800 | |
| }, | |
| { | |
| "epoch": 1.328989384655217, | |
| "grad_norm": 2.8084681034088135, | |
| "learning_rate": 1.7384697890597069e-06, | |
| "loss": 0.3971, | |
| "step": 3850 | |
| }, | |
| { | |
| "epoch": 1.3462501078795202, | |
| "grad_norm": 3.2413413524627686, | |
| "learning_rate": 1.6937790489810514e-06, | |
| "loss": 0.396, | |
| "step": 3900 | |
| }, | |
| { | |
| "epoch": 1.3635108311038233, | |
| "grad_norm": 2.3999781608581543, | |
| "learning_rate": 1.6490883089023956e-06, | |
| "loss": 0.3836, | |
| "step": 3950 | |
| }, | |
| { | |
| "epoch": 1.3807715543281263, | |
| "grad_norm": 3.1420865058898926, | |
| "learning_rate": 1.6043975688237399e-06, | |
| "loss": 0.3757, | |
| "step": 4000 | |
| }, | |
| { | |
| "epoch": 1.3807715543281263, | |
| "eval_gen_len": 54.231239092495635, | |
| "eval_loss": 0.5294080376625061, | |
| "eval_rouge1": 50.76, | |
| "eval_rouge2": 36.73, | |
| "eval_rougeL": 44.34, | |
| "eval_rougeLsum": 44.9, | |
| "eval_runtime": 5340.6521, | |
| "eval_samples_per_second": 1.073, | |
| "eval_steps_per_second": 0.268, | |
| "step": 4000 | |
| } | |
| ], | |
| "logging_steps": 50, | |
| "max_steps": 5794, | |
| "num_input_tokens_seen": 0, | |
| "num_train_epochs": 2, | |
| "save_steps": 500, | |
| "stateful_callbacks": { | |
| "EarlyStoppingCallback": { | |
| "args": { | |
| "early_stopping_patience": 2, | |
| "early_stopping_threshold": 0.0 | |
| }, | |
| "attributes": { | |
| "early_stopping_patience_counter": 2 | |
| } | |
| }, | |
| "TrainerControl": { | |
| "args": { | |
| "should_epoch_stop": false, | |
| "should_evaluate": false, | |
| "should_log": false, | |
| "should_save": true, | |
| "should_training_stop": true | |
| }, | |
| "attributes": {} | |
| } | |
| }, | |
| "total_flos": 1.373273785100206e+17, | |
| "train_batch_size": 4, | |
| "trial_name": null, | |
| "trial_params": null | |
| } | |