Upload folder using huggingface_hub

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+exported/airflow_surrogate.onnx.data filter=lfs diff=lfs merge=lfs -text

best_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d620dcdf22f2456f3048b399aed6c5d6fd0bcd04c9e5755f3ebf3a73afd97ce7
+size 4011061

checkpoint_epoch100.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6bf762695711db3816a4684e219922ae5fa1967a3629347cb6629fae7d9b3628
+size 4011619

checkpoint_epoch200.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dccb6fb0c41f58c394b4e2b75ac2ab248f9b04669717b8e35d2a34ea82e8aae5
+size 4011619

checkpoint_epoch300.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:064ebaace19144bdf947e6d1e49069bc7a05d7a996a8ff169580f34255b76652
+size 4011619

exported/airflow_surrogate.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:af4a1ffd7c1cb6b4e563656de0eaf671dcab819e910e877d321dfd1ccb850cd3
+size 3834

exported/airflow_surrogate.onnx.data

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a8322147c50ba0fa62db9045574a73ab7ef7db0db09052d0f9784d71f55be63c
+size 1331200

exported/airflow_surrogate.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d9f7300a9f6b7f3d488c8a8c0600b7ae124b35494e40472f11322a6cc43a16e6
+size 1344879

final_model.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:88181c9f68585ba2b28dedfcbf5b59c36fb372a52790d8a27a41c3f657c96f2a
+size 1336777

training_history.json

@@ -0,0 +1,2102 @@
+[
+  {
+    "epoch": 1,
+    "train_data_loss": 0.022473466683327586,
+    "train_physics_loss": 0.01187010164993876,
+    "val_loss": 0.015106050823364527,
+    "lr": 0.0009999726120980734
+  },
+  {
+    "epoch": 2,
+    "train_data_loss": 0.013899651029106562,
+    "train_physics_loss": 0.013630859387695676,
+    "val_loss": 0.013837685740943398,
+    "lr": 0.0009998904513956854
+  },
+  {
+    "epoch": 3,
+    "train_data_loss": 0.013421784729415824,
+    "train_physics_loss": 0.011418472630885452,
+    "val_loss": 0.013430576606263076,
+    "lr": 0.0009997535269026829
+  },
+  {
+    "epoch": 4,
+    "train_data_loss": 0.013266340073063772,
+    "train_physics_loss": 0.01067624489053999,
+    "val_loss": 0.013373860503516851,
+    "lr": 0.0009995618536343797
+  },
+  {
+    "epoch": 5,
+    "train_data_loss": 0.013123320399711097,
+    "train_physics_loss": 0.010723257187287371,
+    "val_loss": 0.01327549547497784,
+    "lr": 0.0009993154526099096
+  },
+  {
+    "epoch": 6,
+    "train_data_loss": 0.012946649109927029,
+    "train_physics_loss": 0.010972590213977295,
+    "val_loss": 0.013148192736891008,
+    "lr": 0.0009990143508499217
+  },
+  {
+    "epoch": 7,
+    "train_data_loss": 0.01281524102859185,
+    "train_physics_loss": 0.011043666144461853,
+    "val_loss": 0.012985048853161355,
+    "lr": 0.0009986585813736167
+  },
+  {
+    "epoch": 8,
+    "train_data_loss": 0.012749702496492998,
+    "train_physics_loss": 0.010967634625351181,
+    "val_loss": 0.012940381525925571,
+    "lr": 0.0009982481831951274
+  },
+  {
+    "epoch": 9,
+    "train_data_loss": 0.012709136644070808,
+    "train_physics_loss": 0.010832836469759072,
+    "val_loss": 0.013041857231949125,
+    "lr": 0.0009977832013192385
+  },
+  {
+    "epoch": 10,
+    "train_data_loss": 0.012685124096116033,
+    "train_physics_loss": 0.010778038485056988,
+    "val_loss": 0.012717244320458943,
+    "lr": 0.0009972636867364526
+  },
+  {
+    "epoch": 11,
+    "train_data_loss": 0.012667890659319609,
+    "train_physics_loss": 0.010710710274370164,
+    "val_loss": 0.01279443084833122,
+    "lr": 0.0009966896964173982
+  },
+  {
+    "epoch": 12,
+    "train_data_loss": 0.01266755408605656,
+    "train_physics_loss": 0.010628221061723961,
+    "val_loss": 0.012844510361432068,
+    "lr": 0.000996061293306582
+  },
+  {
+    "epoch": 13,
+    "train_data_loss": 0.012655769386443626,
+    "train_physics_loss": 0.010521324901033963,
+    "val_loss": 0.012777122638879283,
+    "lr": 0.0009953785463154864
+  },
+  {
+    "epoch": 14,
+    "train_data_loss": 0.012648267333927711,
+    "train_physics_loss": 0.010508642736928399,
+    "val_loss": 0.012692450958814832,
+    "lr": 0.0009946415303150131
+  },
+  {
+    "epoch": 15,
+    "train_data_loss": 0.012647806832909697,
+    "train_physics_loss": 0.010481279636724398,
+    "val_loss": 0.012810099607093199,
+    "lr": 0.0009938503261272718
+  },
+  {
+    "epoch": 16,
+    "train_data_loss": 0.012637456417776489,
+    "train_physics_loss": 0.010460906100072376,
+    "val_loss": 0.012817386072129011,
+    "lr": 0.0009930050205167178
+  },
+  {
+    "epoch": 17,
+    "train_data_loss": 0.012633499765390462,
+    "train_physics_loss": 0.010437712354313377,
+    "val_loss": 0.01293633015255534,
+    "lr": 0.0009921057061806368
+  },
+  {
+    "epoch": 18,
+    "train_data_loss": 0.012629570757556685,
+    "train_physics_loss": 0.010397240766933793,
+    "val_loss": 0.012809061047230516,
+    "lr": 0.0009911524817389807
+  },
+  {
+    "epoch": 19,
+    "train_data_loss": 0.01262768653824049,
+    "train_physics_loss": 0.010390486472782871,
+    "val_loss": 0.012760479423788286,
+    "lr": 0.0009901454517235507
+  },
+  {
+    "epoch": 20,
+    "train_data_loss": 0.012616622535621418,
+    "train_physics_loss": 0.010348897434389343,
+    "val_loss": 0.012818439908686185,
+    "lr": 0.0009890847265665364
+  },
+  {
+    "epoch": 21,
+    "train_data_loss": 0.012626901097539141,
+    "train_physics_loss": 0.010304349133536304,
+    "val_loss": 0.012867256630993178,
+    "lr": 0.0009879704225884047
+  },
+  {
+    "epoch": 22,
+    "train_data_loss": 0.012615944781964485,
+    "train_physics_loss": 0.010298519649770839,
+    "val_loss": 0.012950033565322238,
+    "lr": 0.000986802661985144
+  },
+  {
+    "epoch": 23,
+    "train_data_loss": 0.012614420489664322,
+    "train_physics_loss": 0.010303166979460834,
+    "val_loss": 0.012858719422271656,
+    "lr": 0.0009855815728148636
+  },
+  {
+    "epoch": 24,
+    "train_data_loss": 0.012611826164276821,
+    "train_physics_loss": 0.010265776322910694,
+    "val_loss": 0.012922665269504632,
+    "lr": 0.0009843072889837517
+  },
+  {
+    "epoch": 25,
+    "train_data_loss": 0.012614877924563424,
+    "train_physics_loss": 0.010234038684821107,
+    "val_loss": 0.012785898078413259,
+    "lr": 0.00098297995023139
+  },
+  {
+    "epoch": 26,
+    "train_data_loss": 0.012606659287364139,
+    "train_physics_loss": 0.010236529976242633,
+    "val_loss": 0.012814379590112836,
+    "lr": 0.0009815997021154308
+  },
+  {
+    "epoch": 27,
+    "train_data_loss": 0.012606766458381624,
+    "train_physics_loss": 0.010214985611599356,
+    "val_loss": 0.012839672350955586,
+    "lr": 0.0009801666959956335
+  },
+  {
+    "epoch": 28,
+    "train_data_loss": 0.012613246514262239,
+    "train_physics_loss": 0.01022584195932695,
+    "val_loss": 0.012769114286188156,
+    "lr": 0.000978681089017268
+  },
+  {
+    "epoch": 29,
+    "train_data_loss": 0.01260697307768481,
+    "train_physics_loss": 0.010184626020773878,
+    "val_loss": 0.01284450313617145,
+    "lr": 0.0009771430440938809
+  },
+  {
+    "epoch": 30,
+    "train_data_loss": 0.01260651248123716,
+    "train_physics_loss": 0.010168569733298594,
+    "val_loss": 0.012909238497095724,
+    "lr": 0.0009755527298894299
+  },
+  {
+    "epoch": 31,
+    "train_data_loss": 0.01260541947081374,
+    "train_physics_loss": 0.01012271784238899,
+    "val_loss": 0.012809283032472576,
+    "lr": 0.0009739103207997887
+  },
+  {
+    "epoch": 32,
+    "train_data_loss": 0.012602058944400167,
+    "train_physics_loss": 0.01017516634411106,
+    "val_loss": 0.01271106191580334,
+    "lr": 0.0009722159969336225
+  },
+  {
+    "epoch": 33,
+    "train_data_loss": 0.01260634980374767,
+    "train_physics_loss": 0.010135488759591978,
+    "val_loss": 0.012796789497858094,
+    "lr": 0.0009704699440926363
+  },
+  {
+    "epoch": 34,
+    "train_data_loss": 0.012602878554213432,
+    "train_physics_loss": 0.010119629371734567,
+    "val_loss": 0.01281063037083274,
+    "lr": 0.0009686723537512004
+  },
+  {
+    "epoch": 35,
+    "train_data_loss": 0.012594679731034463,
+    "train_physics_loss": 0.010145070748774784,
+    "val_loss": 0.012884211403528047,
+    "lr": 0.0009668234230353527
+  },
+  {
+    "epoch": 36,
+    "train_data_loss": 0.012600851322981959,
+    "train_physics_loss": 0.010098078858948523,
+    "val_loss": 0.01279298850755778,
+    "lr": 0.0009649233547011821
+  },
+  {
+    "epoch": 37,
+    "train_data_loss": 0.01260253784036444,
+    "train_physics_loss": 0.010102063581785736,
+    "val_loss": 0.012805103489588346,
+    "lr": 0.000962972357112593
+  },
+  {
+    "epoch": 38,
+    "train_data_loss": 0.01259780236019751,
+    "train_physics_loss": 0.010106293678962302,
+    "val_loss": 0.012759791003660328,
+    "lr": 0.0009609706442184566
+  },
+  {
+    "epoch": 39,
+    "train_data_loss": 0.012601664837668924,
+    "train_physics_loss": 0.010113950755157326,
+    "val_loss": 0.012771490961313248,
+    "lr": 0.0009589184355291491
+  },
+  {
+    "epoch": 40,
+    "train_data_loss": 0.012595727566127759,
+    "train_physics_loss": 0.01007684862216408,
+    "val_loss": 0.012719864438798639,
+    "lr": 0.0009568159560924797
+  },
+  {
+    "epoch": 41,
+    "train_data_loss": 0.012597967214896512,
+    "train_physics_loss": 0.010072134905256176,
+    "val_loss": 0.01282208536060587,
+    "lr": 0.0009546634364690113
+  },
+  {
+    "epoch": 42,
+    "train_data_loss": 0.01259520038099963,
+    "train_physics_loss": 0.010069370020598004,
+    "val_loss": 0.012831116568357233,
+    "lr": 0.0009524611127067773
+  },
+  {
+    "epoch": 43,
+    "train_data_loss": 0.012597076692026412,
+    "train_physics_loss": 0.010058320841382645,
+    "val_loss": 0.012734376570029605,
+    "lr": 0.0009502092263153963
+  },
+  {
+    "epoch": 44,
+    "train_data_loss": 0.01259513001401239,
+    "train_physics_loss": 0.010075274289389488,
+    "val_loss": 0.012829494489837558,
+    "lr": 0.0009479080242395872
+  },
+  {
+    "epoch": 45,
+    "train_data_loss": 0.012592901526976808,
+    "train_physics_loss": 0.010048788532162302,
+    "val_loss": 0.012752350983059695,
+    "lr": 0.0009455577588320901
+  },
+  {
+    "epoch": 46,
+    "train_data_loss": 0.012588654418583726,
+    "train_physics_loss": 0.010061453218506454,
+    "val_loss": 0.012717082120117641,
+    "lr": 0.0009431586878259921
+  },
+  {
+    "epoch": 47,
+    "train_data_loss": 0.012595284417398853,
+    "train_physics_loss": 0.010037362374068437,
+    "val_loss": 0.012752050834317361,
+    "lr": 0.0009407110743064639
+  },
+  {
+    "epoch": 48,
+    "train_data_loss": 0.012592789810681026,
+    "train_physics_loss": 0.010049911744298813,
+    "val_loss": 0.012788701314298857,
+    "lr": 0.0009382151866819102
+  },
+  {
+    "epoch": 49,
+    "train_data_loss": 0.012599430640760025,
+    "train_physics_loss": 0.010004244569823344,
+    "val_loss": 0.01277322783285091,
+    "lr": 0.0009356712986545351
+  },
+  {
+    "epoch": 50,
+    "train_data_loss": 0.01259153186956834,
+    "train_physics_loss": 0.010035852062210186,
+    "val_loss": 0.012886076264323728,
+    "lr": 0.0009330796891903276
+  },
+  {
+    "epoch": 51,
+    "train_data_loss": 0.012597918066865919,
+    "train_physics_loss": 0.00997185059054357,
+    "val_loss": 0.01281377652870311,
+    "lr": 0.0009304406424884703
+  },
+  {
+    "epoch": 52,
+    "train_data_loss": 0.012592370706686371,
+    "train_physics_loss": 0.010032874470191392,
+    "val_loss": 0.012828720553267387,
+    "lr": 0.0009277544479501735
+  },
+  {
+    "epoch": 53,
+    "train_data_loss": 0.01258598788123599,
+    "train_physics_loss": 0.01004250393031727,
+    "val_loss": 0.012794583742957442,
+    "lr": 0.000925021400146939
+  },
+  {
+    "epoch": 54,
+    "train_data_loss": 0.012586580357908411,
+    "train_physics_loss": 0.010007595337672863,
+    "val_loss": 0.012816229812620629,
+    "lr": 0.000922241798788257
+  },
+  {
+    "epoch": 55,
+    "train_data_loss": 0.012585891785358115,
+    "train_physics_loss": 0.010037188338684855,
+    "val_loss": 0.012720129557795102,
+    "lr": 0.0009194159486887398
+  },
+  {
+    "epoch": 56,
+    "train_data_loss": 0.012591805395650796,
+    "train_physics_loss": 0.00998884854764029,
+    "val_loss": 0.012762742327345956,
+    "lr": 0.0009165441597346951
+  },
+  {
+    "epoch": 57,
+    "train_data_loss": 0.012589895825628085,
+    "train_physics_loss": 0.00999864139744351,
+    "val_loss": 0.012790337302571824,
+    "lr": 0.000913626746850144
+  },
+  {
+    "epoch": 58,
+    "train_data_loss": 0.012588737608486718,
+    "train_physics_loss": 0.010002488245660938,
+    "val_loss": 0.012766058211244883,
+    "lr": 0.0009106640299622855
+  },
+  {
+    "epoch": 59,
+    "train_data_loss": 0.012581853261411868,
+    "train_physics_loss": 0.010009701271847145,
+    "val_loss": 0.012760821542131805,
+    "lr": 0.0009076563339664131
+  },
+  {
+    "epoch": 60,
+    "train_data_loss": 0.012589706926194833,
+    "train_physics_loss": 0.00998485304832289,
+    "val_loss": 0.01283443680093173,
+    "lr": 0.0009046039886902866
+  },
+  {
+    "epoch": 61,
+    "train_data_loss": 0.012585796262058068,
+    "train_physics_loss": 0.010006735110178279,
+    "val_loss": 0.012763994092482233,
+    "lr": 0.000901507328857962
+  },
+  {
+    "epoch": 62,
+    "train_data_loss": 0.012589833462359897,
+    "train_physics_loss": 0.009954824495841475,
+    "val_loss": 0.012876752193175977,
+    "lr": 0.0008983666940530862
+  },
+  {
+    "epoch": 63,
+    "train_data_loss": 0.012590740360248926,
+    "train_physics_loss": 0.00995652463138669,
+    "val_loss": 0.012737033160163029,
+    "lr": 0.0008951824286816575
+  },
+  {
+    "epoch": 64,
+    "train_data_loss": 0.012592553415563685,
+    "train_physics_loss": 0.009945004966385677,
+    "val_loss": 0.012721124593348752,
+    "lr": 0.0008919548819342572
+  },
+  {
+    "epoch": 65,
+    "train_data_loss": 0.012592435183544312,
+    "train_physics_loss": 0.0099332054398346,
+    "val_loss": 0.012728127808640561,
+    "lr": 0.0008886844077477572
+  },
+  {
+    "epoch": 66,
+    "train_data_loss": 0.012586160246600914,
+    "train_physics_loss": 0.009959203943935359,
+    "val_loss": 0.012812184072249839,
+    "lr": 0.0008853713647665069
+  },
+  {
+    "epoch": 67,
+    "train_data_loss": 0.012586846721466398,
+    "train_physics_loss": 0.009942993887503979,
+    "val_loss": 0.012896400793725926,
+    "lr": 0.000882016116303004
+  },
+  {
+    "epoch": 68,
+    "train_data_loss": 0.01259083994814465,
+    "train_physics_loss": 0.009957250707784571,
+    "val_loss": 0.012827425526695387,
+    "lr": 0.0008786190302980524
+  },
+  {
+    "epoch": 69,
+    "train_data_loss": 0.01258644910504515,
+    "train_physics_loss": 0.009949009015491724,
+    "val_loss": 0.012783901623239922,
+    "lr": 0.0008751804792804147
+  },
+  {
+    "epoch": 70,
+    "train_data_loss": 0.01258729673406035,
+    "train_physics_loss": 0.00993573876447707,
+    "val_loss": 0.012823309118468915,
+    "lr": 0.0008717008403259585
+  },
+  {
+    "epoch": 71,
+    "train_data_loss": 0.012583347553461507,
+    "train_physics_loss": 0.009960084766559515,
+    "val_loss": 0.012769305597870581,
+    "lr": 0.0008681804950163073
+  },
+  {
+    "epoch": 72,
+    "train_data_loss": 0.012582547441336982,
+    "train_physics_loss": 0.009962834670531795,
+    "val_loss": 0.012753026387203605,
+    "lr": 0.0008646198293969952
+  },
+  {
+    "epoch": 73,
+    "train_data_loss": 0.012583516091037633,
+    "train_physics_loss": 0.00992259154877355,
+    "val_loss": 0.012747570481752195,
+    "lr": 0.000861019233935132
+  },
+  {
+    "epoch": 74,
+    "train_data_loss": 0.012583379112396114,
+    "train_physics_loss": 0.009939931561035833,
+    "val_loss": 0.012747265901716967,
+    "lr": 0.0008573791034765854
+  },
+  {
+    "epoch": 75,
+    "train_data_loss": 0.012587265254650442,
+    "train_physics_loss": 0.009926732975911822,
+    "val_loss": 0.012865305818136661,
+    "lr": 0.0008536998372026806
+  },
+  {
+    "epoch": 76,
+    "train_data_loss": 0.012586572654615996,
+    "train_physics_loss": 0.009897622315167137,
+    "val_loss": 0.012796224508013937,
+    "lr": 0.0008499818385864263
+  },
+  {
+    "epoch": 77,
+    "train_data_loss": 0.01258558461149232,
+    "train_physics_loss": 0.009944508483092631,
+    "val_loss": 0.012767568343288956,
+    "lr": 0.0008462255153482683
+  },
+  {
+    "epoch": 78,
+    "train_data_loss": 0.012587216786997821,
+    "train_physics_loss": 0.009901328675053039,
+    "val_loss": 0.0127644557731166,
+    "lr": 0.0008424312794113801
+  },
+  {
+    "epoch": 79,
+    "train_data_loss": 0.012581497888510209,
+    "train_physics_loss": 0.009931052464150613,
+    "val_loss": 0.012788614213106132,
+    "lr": 0.0008385995468564891
+  },
+  {
+    "epoch": 80,
+    "train_data_loss": 0.0125793327084359,
+    "train_physics_loss": 0.009944459709711726,
+    "val_loss": 0.012830212486967925,
+    "lr": 0.0008347307378762498
+  },
+  {
+    "epoch": 81,
+    "train_data_loss": 0.012587913029263436,
+    "train_physics_loss": 0.009899264106092235,
+    "val_loss": 0.012825400328203555,
+    "lr": 0.0008308252767291642
+  },
+  {
+    "epoch": 82,
+    "train_data_loss": 0.012584351363681977,
+    "train_physics_loss": 0.00988684922626394,
+    "val_loss": 0.012842707726503573,
+    "lr": 0.0008268835916930579
+  },
+  {
+    "epoch": 83,
+    "train_data_loss": 0.012586653960667254,
+    "train_physics_loss": 0.009876643832370045,
+    "val_loss": 0.012821952970097623,
+    "lr": 0.0008229061150181134
+  },
+  {
+    "epoch": 84,
+    "train_data_loss": 0.012583348901846942,
+    "train_physics_loss": 0.009901542478087279,
+    "val_loss": 0.012723073490985459,
+    "lr": 0.0008188932828794706
+  },
+  {
+    "epoch": 85,
+    "train_data_loss": 0.012581489197344323,
+    "train_physics_loss": 0.009909631062362633,
+    "val_loss": 0.012789643692573714,
+    "lr": 0.0008148455353293939
+  },
+  {
+    "epoch": 86,
+    "train_data_loss": 0.012583387937870378,
+    "train_physics_loss": 0.009891553773090554,
+    "val_loss": 0.01286664588617221,
+    "lr": 0.0008107633162490162
+  },
+  {
+    "epoch": 87,
+    "train_data_loss": 0.012577128794532008,
+    "train_physics_loss": 0.009914282571517313,
+    "val_loss": 0.012819094913861444,
+    "lr": 0.0008066470732996619
+  },
+  {
+    "epoch": 88,
+    "train_data_loss": 0.01258433815728046,
+    "train_physics_loss": 0.009895905977790111,
+    "val_loss": 0.012753312438425998,
+    "lr": 0.0008024972578737564
+  },
+  {
+    "epoch": 89,
+    "train_data_loss": 0.01258163837860895,
+    "train_physics_loss": 0.00988387821535571,
+    "val_loss": 0.012768999035019548,
+    "lr": 0.0007983143250453251
+  },
+  {
+    "epoch": 90,
+    "train_data_loss": 0.012576124906554055,
+    "train_physics_loss": 0.009918274128176908,
+    "val_loss": 0.012736779059313478,
+    "lr": 0.0007940987335200903
+  },
+  {
+    "epoch": 91,
+    "train_data_loss": 0.012579962635939442,
+    "train_physics_loss": 0.009869470868959486,
+    "val_loss": 0.01268671683576559,
+    "lr": 0.000789850945585168
+  },
+  {
+    "epoch": 92,
+    "train_data_loss": 0.012579509871790939,
+    "train_physics_loss": 0.009907380366729718,
+    "val_loss": 0.012849942578243153,
+    "lr": 0.0007855714270583737
+  },
+  {
+    "epoch": 93,
+    "train_data_loss": 0.012579689438559073,
+    "train_physics_loss": 0.009876980161757351,
+    "val_loss": 0.012814963461771127,
+    "lr": 0.0007812606472371393
+  },
+  {
+    "epoch": 94,
+    "train_data_loss": 0.01258025295763812,
+    "train_physics_loss": 0.009876140421592194,
+    "val_loss": 0.012812738209181735,
+    "lr": 0.0007769190788470504
+  },
+  {
+    "epoch": 95,
+    "train_data_loss": 0.012583254719863468,
+    "train_physics_loss": 0.009876005443438633,
+    "val_loss": 0.01280471581154533,
+    "lr": 0.000772547197990006
+  },
+  {
+    "epoch": 96,
+    "train_data_loss": 0.012579683785236651,
+    "train_physics_loss": 0.00986051177199527,
+    "val_loss": 0.012789500712026511,
+    "lr": 0.0007681454840920087
+  },
+  {
+    "epoch": 97,
+    "train_data_loss": 0.012579624496923451,
+    "train_physics_loss": 0.009871696206817025,
+    "val_loss": 0.012721502237142093,
+    "lr": 0.0007637144198505905
+  },
+  {
+    "epoch": 98,
+    "train_data_loss": 0.012578232227992514,
+    "train_physics_loss": 0.009867192786787786,
+    "val_loss": 0.012760533943712231,
+    "lr": 0.0007592544911818786
+  },
+  {
+    "epoch": 99,
+    "train_data_loss": 0.012579399046171775,
+    "train_physics_loss": 0.00987169238609709,
+    "val_loss": 0.012767564986021288,
+    "lr": 0.0007547661871673105
+  },
+  {
+    "epoch": 100,
+    "train_data_loss": 0.012582293227385638,
+    "train_physics_loss": 0.009848106376812625,
+    "val_loss": 0.012876572575600397,
+    "lr": 0.00075025
+  },
+  {
+    "epoch": 101,
+    "train_data_loss": 0.012575789457240073,
+    "train_physics_loss": 0.009865001066670924,
+    "val_loss": 0.012861734699277629,
+    "lr": 0.0007457064249307629
+  },
+  {
+    "epoch": 102,
+    "train_data_loss": 0.01257885357919918,
+    "train_physics_loss": 0.009867234703372507,
+    "val_loss": 0.012789336183378774,
+    "lr": 0.0007411359602138069
+  },
+  {
+    "epoch": 103,
+    "train_data_loss": 0.012581363237291518,
+    "train_physics_loss": 0.009842817017727264,
+    "val_loss": 0.012739327510878924,
+    "lr": 0.0007365391070520927
+  },
+  {
+    "epoch": 104,
+    "train_data_loss": 0.012576005221883532,
+    "train_physics_loss": 0.009857827330987187,
+    "val_loss": 0.012749136132638781,
+    "lr": 0.0007319163695423712
+  },
+  {
+    "epoch": 105,
+    "train_data_loss": 0.012581131894643325,
+    "train_physics_loss": 0.0098331910796619,
+    "val_loss": 0.012827450109104957,
+    "lr": 0.0007272682546199038
+  },
+  {
+    "epoch": 106,
+    "train_data_loss": 0.012578340894074322,
+    "train_physics_loss": 0.009848801295433822,
+    "val_loss": 0.012800938029202723,
+    "lr": 0.0007225952720028714
+  },
+  {
+    "epoch": 107,
+    "train_data_loss": 0.012576211250936737,
+    "train_physics_loss": 0.009830231513241324,
+    "val_loss": 0.012667190921943515,
+    "lr": 0.0007178979341364778
+  },
+  {
+    "epoch": 108,
+    "train_data_loss": 0.012578077920035687,
+    "train_physics_loss": 0.009851272475942828,
+    "val_loss": 0.012804951060623411,
+    "lr": 0.000713176756136754
+  },
+  {
+    "epoch": 109,
+    "train_data_loss": 0.01258259848912425,
+    "train_physics_loss": 0.009825747778225669,
+    "val_loss": 0.01278666427899753,
+    "lr": 0.0007084322557340706
+  },
+  {
+    "epoch": 110,
+    "train_data_loss": 0.012573414312523614,
+    "train_physics_loss": 0.009859455721117062,
+    "val_loss": 0.012790028809479648,
+    "lr": 0.0007036649532163623
+  },
+  {
+    "epoch": 111,
+    "train_data_loss": 0.01257955084644979,
+    "train_physics_loss": 0.009822184055605355,
+    "val_loss": 0.012722260341228496,
+    "lr": 0.000698875371372073
+  },
+  {
+    "epoch": 112,
+    "train_data_loss": 0.012574612723214577,
+    "train_physics_loss": 0.009845599505619147,
+    "val_loss": 0.012823110927016504,
+    "lr": 0.0006940640354328256
+  },
+  {
+    "epoch": 113,
+    "train_data_loss": 0.012581911317975855,
+    "train_physics_loss": 0.009809892656426502,
+    "val_loss": 0.012773014259554686,
+    "lr": 0.0006892314730158245
+  },
+  {
+    "epoch": 114,
+    "train_data_loss": 0.012577739994852774,
+    "train_physics_loss": 0.009841520234984849,
+    "val_loss": 0.01277352446660159,
+    "lr": 0.0006843782140659969
+  },
+  {
+    "epoch": 115,
+    "train_data_loss": 0.012577877910118515,
+    "train_physics_loss": 0.009811809498541043,
+    "val_loss": 0.012796049854988533,
+    "lr": 0.0006795047907978775
+  },
+  {
+    "epoch": 116,
+    "train_data_loss": 0.01257972791084076,
+    "train_physics_loss": 0.00980332191588191,
+    "val_loss": 0.012757827092202441,
+    "lr": 0.0006746117376372468
+  },
+  {
+    "epoch": 117,
+    "train_data_loss": 0.012578231724336093,
+    "train_physics_loss": 0.009806328405408751,
+    "val_loss": 0.012764481264316747,
+    "lr": 0.0006696995911625233
+  },
+  {
+    "epoch": 118,
+    "train_data_loss": 0.012574184470345218,
+    "train_physics_loss": 0.009831383509784089,
+    "val_loss": 0.01279796655440042,
+    "lr": 0.0006647688900459224
+  },
+  {
+    "epoch": 119,
+    "train_data_loss": 0.012572800752970141,
+    "train_physics_loss": 0.009832751675662664,
+    "val_loss": 0.012830669698756068,
+    "lr": 0.000659820174994386
+  },
+  {
+    "epoch": 120,
+    "train_data_loss": 0.012576608399045874,
+    "train_physics_loss": 0.009809851916807314,
+    "val_loss": 0.012771434781532134,
+    "lr": 0.0006548539886902864
+  },
+  {
+    "epoch": 121,
+    "train_data_loss": 0.012574334719008016,
+    "train_physics_loss": 0.009818665499479087,
+    "val_loss": 0.012764522588000662,
+    "lr": 0.0006498708757319155
+  },
+  {
+    "epoch": 122,
+    "train_data_loss": 0.012578039325816128,
+    "train_physics_loss": 0.009779148579591026,
+    "val_loss": 0.012772347680443236,
+    "lr": 0.0006448713825737638
+  },
+  {
+    "epoch": 123,
+    "train_data_loss": 0.01257394260569587,
+    "train_physics_loss": 0.009816153178192972,
+    "val_loss": 0.012812892132769189,
+    "lr": 0.0006398560574665953
+  },
+  {
+    "epoch": 124,
+    "train_data_loss": 0.01258043720752736,
+    "train_physics_loss": 0.009784862726036127,
+    "val_loss": 0.012759816757733784,
+    "lr": 0.0006348254503973254
+  },
+  {
+    "epoch": 125,
+    "train_data_loss": 0.01258056156294848,
+    "train_physics_loss": 0.009772810376332312,
+    "val_loss": 0.012810045800682518,
+    "lr": 0.0006297801130287095
+  },
+  {
+    "epoch": 126,
+    "train_data_loss": 0.012575073068717626,
+    "train_physics_loss": 0.009799381933639805,
+    "val_loss": 0.012822376308782447,
+    "lr": 0.0006247205986388451
+  },
+  {
+    "epoch": 127,
+    "train_data_loss": 0.012573685489976654,
+    "train_physics_loss": 0.009794473335274143,
+    "val_loss": 0.012742571239810317,
+    "lr": 0.0006196474620605014
+  },
+  {
+    "epoch": 128,
+    "train_data_loss": 0.012578182776000852,
+    "train_physics_loss": 0.009786415884034456,
+    "val_loss": 0.012806085005943333,
+    "lr": 0.0006145612596202728
+  },
+  {
+    "epoch": 129,
+    "train_data_loss": 0.012574072052464878,
+    "train_physics_loss": 0.009799356908100946,
+    "val_loss": 0.012769186471198355,
+    "lr": 0.0006094625490775733
+  },
+  {
+    "epoch": 130,
+    "train_data_loss": 0.012575308255059555,
+    "train_physics_loss": 0.009781143571509922,
+    "val_loss": 0.012737738148820016,
+    "lr": 0.000604351889563471
+  },
+  {
+    "epoch": 131,
+    "train_data_loss": 0.012574792862560536,
+    "train_physics_loss": 0.009791389183694429,
+    "val_loss": 0.01273264036694121,
+    "lr": 0.0005992298415193736
+  },
+  {
+    "epoch": 132,
+    "train_data_loss": 0.012571838145379323,
+    "train_physics_loss": 0.009789039161713683,
+    "val_loss": 0.01278964030526338,
+    "lr": 0.0005940969666355697
+  },
+  {
+    "epoch": 133,
+    "train_data_loss": 0.012571644055634115,
+    "train_physics_loss": 0.009802120728895379,
+    "val_loss": 0.01277148102470223,
+    "lr": 0.000588953827789632
+  },
+  {
+    "epoch": 134,
+    "train_data_loss": 0.012574980604467626,
+    "train_physics_loss": 0.009765819740080516,
+    "val_loss": 0.01278780001185594,
+    "lr": 0.0005838009889846932
+  },
+  {
+    "epoch": 135,
+    "train_data_loss": 0.012570385884783413,
+    "train_physics_loss": 0.009799905499510566,
+    "val_loss": 0.012849363678645703,
+    "lr": 0.0005786390152875954
+  },
+  {
+    "epoch": 136,
+    "train_data_loss": 0.012574306408215519,
+    "train_physics_loss": 0.009765208589241785,
+    "val_loss": 0.01277979772778288,
+    "lr": 0.0005734684727669247
+  },
+  {
+    "epoch": 137,
+    "train_data_loss": 0.012572416995052934,
+    "train_physics_loss": 0.009780715392863615,
+    "val_loss": 0.012823967157953208,
+    "lr": 0.0005682899284309351
+  },
+  {
+    "epoch": 138,
+    "train_data_loss": 0.012580577197504111,
+    "train_physics_loss": 0.00973632786620896,
+    "val_loss": 0.012763977118377243,
+    "lr": 0.00056310395016537
+  },
+  {
+    "epoch": 139,
+    "train_data_loss": 0.012572099702111119,
+    "train_physics_loss": 0.009777218082202227,
+    "val_loss": 0.012765344690471407,
+    "lr": 0.000557911106671187
+  },
+  {
+    "epoch": 140,
+    "train_data_loss": 0.01257405594429549,
+    "train_physics_loss": 0.009747151318987022,
+    "val_loss": 0.012739845994679679,
+    "lr": 0.0005527119674021931
+  },
+  {
+    "epoch": 141,
+    "train_data_loss": 0.012572883041593224,
+    "train_physics_loss": 0.009767118804298158,
+    "val_loss": 0.012741284925610788,
+    "lr": 0.000547507102502598
+  },
+  {
+    "epoch": 142,
+    "train_data_loss": 0.012574377114508139,
+    "train_physics_loss": 0.009756009820742671,
+    "val_loss": 0.012771151276425488,
+    "lr": 0.0005422970827444917
+  },
+  {
+    "epoch": 143,
+    "train_data_loss": 0.012569193942101223,
+    "train_physics_loss": 0.009782360013100748,
+    "val_loss": 0.012737751577890689,
+    "lr": 0.0005370824794652521
+  },
+  {
+    "epoch": 144,
+    "train_data_loss": 0.01257071845000168,
+    "train_physics_loss": 0.009777486015047588,
+    "val_loss": 0.012778979365623767,
+    "lr": 0.0005318638645048922
+  },
+  {
+    "epoch": 145,
+    "train_data_loss": 0.012572385699433438,
+    "train_physics_loss": 0.009769226530428969,
+    "val_loss": 0.012785036582499743,
+    "lr": 0.0005266418101433506
+  },
+  {
+    "epoch": 146,
+    "train_data_loss": 0.012570336641323182,
+    "train_physics_loss": 0.00975376602962076,
+    "val_loss": 0.012742606194449528,
+    "lr": 0.0005214168890377354
+  },
+  {
+    "epoch": 147,
+    "train_data_loss": 0.012569126429167033,
+    "train_physics_loss": 0.009758648114452099,
+    "val_loss": 0.012761418549944796,
+    "lr": 0.0005161896741595251
+  },
+  {
+    "epoch": 148,
+    "train_data_loss": 0.012579386679197625,
+    "train_physics_loss": 0.0097272920332128,
+    "val_loss": 0.012826586983376933,
+    "lr": 0.0005109607387317369
+  },
+  {
+    "epoch": 149,
+    "train_data_loss": 0.01257193935911332,
+    "train_physics_loss": 0.009737900085393799,
+    "val_loss": 0.012798442580406704,
+    "lr": 0.0005057306561660649
+  },
+  {
+    "epoch": 150,
+    "train_data_loss": 0.012573963858229611,
+    "train_physics_loss": 0.009735270928188565,
+    "val_loss": 0.012794722705298373,
+    "lr": 0.0005005000000000001
+  },
+  {
+    "epoch": 151,
+    "train_data_loss": 0.012571401401039432,
+    "train_physics_loss": 0.009752343385789607,
+    "val_loss": 0.012752815232341808,
+    "lr": 0.0004952693438339353
+  },
+  {
+    "epoch": 152,
+    "train_data_loss": 0.012573004375075818,
+    "train_physics_loss": 0.009728735372904808,
+    "val_loss": 0.01275747336988007,
+    "lr": 0.0004900392612682634
+  },
+  {
+    "epoch": 153,
+    "train_data_loss": 0.012571861857877058,
+    "train_physics_loss": 0.009735487489846218,
+    "val_loss": 0.01275337127698285,
+    "lr": 0.00048481032584047495
+  },
+  {
+    "epoch": 154,
+    "train_data_loss": 0.012570184434585598,
+    "train_physics_loss": 0.009745850906383946,
+    "val_loss": 0.012746193431197636,
+    "lr": 0.0004795831109622648
+  },
+  {
+    "epoch": 155,
+    "train_data_loss": 0.01257196238416422,
+    "train_physics_loss": 0.00973255441398616,
+    "val_loss": 0.012789492442783329,
+    "lr": 0.0004743581898566495
+  },
+  {
+    "epoch": 156,
+    "train_data_loss": 0.012575724877883632,
+    "train_physics_loss": 0.00972171756218009,
+    "val_loss": 0.012767972357030357,
+    "lr": 0.00046913613549510807
+  },
+  {
+    "epoch": 157,
+    "train_data_loss": 0.01256797264066774,
+    "train_physics_loss": 0.00974127149962214,
+    "val_loss": 0.012776874486477144,
+    "lr": 0.00046391752053474806
+  },
+  {
+    "epoch": 158,
+    "train_data_loss": 0.012572832278327652,
+    "train_physics_loss": 0.009713378093387416,
+    "val_loss": 0.012774635699667757,
+    "lr": 0.0004587029172555084
+  },
+  {
+    "epoch": 159,
+    "train_data_loss": 0.012572736345033605,
+    "train_physics_loss": 0.009714199362616046,
+    "val_loss": 0.012792233723185716,
+    "lr": 0.0004534928974974021
+  },
+  {
+    "epoch": 160,
+    "train_data_loss": 0.012574268800102217,
+    "train_physics_loss": 0.00970782342930616,
+    "val_loss": 0.012793564891082144,
+    "lr": 0.0004482880325978072
+  },
+  {
+    "epoch": 161,
+    "train_data_loss": 0.012566329600321275,
+    "train_physics_loss": 0.009741611833519796,
+    "val_loss": 0.012808131151682427,
+    "lr": 0.0004430888933288132
+  },
+  {
+    "epoch": 162,
+    "train_data_loss": 0.012572896497172128,
+    "train_physics_loss": 0.009704010145812944,
+    "val_loss": 0.01280406755094807,
+    "lr": 0.0004378960498346301
+  },
+  {
+    "epoch": 163,
+    "train_data_loss": 0.012567609106764847,
+    "train_physics_loss": 0.009733467145453594,
+    "val_loss": 0.012755750821182323,
+    "lr": 0.000432710071569065
+  },
+  {
+    "epoch": 164,
+    "train_data_loss": 0.012574743147253764,
+    "train_physics_loss": 0.00969284999606675,
+    "val_loss": 0.01281561061078983,
+    "lr": 0.0004275315272330756
+  },
+  {
+    "epoch": 165,
+    "train_data_loss": 0.012572162915409968,
+    "train_physics_loss": 0.009710618148096598,
+    "val_loss": 0.012840515580929576,
+    "lr": 0.0004223609847124048
+  },
+  {
+    "epoch": 166,
+    "train_data_loss": 0.012573791572389385,
+    "train_physics_loss": 0.009692555208614475,
+    "val_loss": 0.012832180702037389,
+    "lr": 0.0004171990110153068
+  },
+  {
+    "epoch": 167,
+    "train_data_loss": 0.012567507242695543,
+    "train_physics_loss": 0.00971275540793304,
+    "val_loss": 0.012776570274464546,
+    "lr": 0.00041204617221036815
+  },
+  {
+    "epoch": 168,
+    "train_data_loss": 0.012575457939980605,
+    "train_physics_loss": 0.009681873551907418,
+    "val_loss": 0.012782315445703364,
+    "lr": 0.00040690303336443054
+  },
+  {
+    "epoch": 169,
+    "train_data_loss": 0.012566837229442551,
+    "train_physics_loss": 0.009720406457923169,
+    "val_loss": 0.012803449746101134,
+    "lr": 0.00040177015848062643
+  },
+  {
+    "epoch": 170,
+    "train_data_loss": 0.012569762798171116,
+    "train_physics_loss": 0.009700655511295887,
+    "val_loss": 0.012766294421688203,
+    "lr": 0.00039664811043652916
+  },
+  {
+    "epoch": 171,
+    "train_data_loss": 0.012568188168504319,
+    "train_physics_loss": 0.009711412951505728,
+    "val_loss": 0.012804981779247042,
+    "lr": 0.000391537450922427
+  },
+  {
+    "epoch": 172,
+    "train_data_loss": 0.012568733349188004,
+    "train_physics_loss": 0.009704524857933992,
+    "val_loss": 0.012820275132394126,
+    "lr": 0.00038643874037972754
+  },
+  {
+    "epoch": 173,
+    "train_data_loss": 0.012567494181205697,
+    "train_physics_loss": 0.009705375805779924,
+    "val_loss": 0.01278548842416175,
+    "lr": 0.00038135253793949894
+  },
+  {
+    "epoch": 174,
+    "train_data_loss": 0.012562548405927318,
+    "train_physics_loss": 0.00971493521878457,
+    "val_loss": 0.01278641730576994,
+    "lr": 0.0003762794013611551
+  },
+  {
+    "epoch": 175,
+    "train_data_loss": 0.012568520592345346,
+    "train_physics_loss": 0.009692138928630207,
+    "val_loss": 0.012749687948774907,
+    "lr": 0.00037121988697129095
+  },
+  {
+    "epoch": 176,
+    "train_data_loss": 0.012569242413288276,
+    "train_physics_loss": 0.009687225885717535,
+    "val_loss": 0.012787931080486986,
+    "lr": 0.00036617454960267493
+  },
+  {
+    "epoch": 177,
+    "train_data_loss": 0.012563498595294735,
+    "train_physics_loss": 0.009711011631000766,
+    "val_loss": 0.012739202848845912,
+    "lr": 0.0003611439425334049
+  },
+  {
+    "epoch": 178,
+    "train_data_loss": 0.01256334726332883,
+    "train_physics_loss": 0.009722407795912966,
+    "val_loss": 0.012809885327794378,
+    "lr": 0.0003561286174262363
+  },
+  {
+    "epoch": 179,
+    "train_data_loss": 0.01256792999599023,
+    "train_physics_loss": 0.009682706908890146,
+    "val_loss": 0.012762185787000963,
+    "lr": 0.0003511291242680847
+  },
+  {
+    "epoch": 180,
+    "train_data_loss": 0.012572666511745448,
+    "train_physics_loss": 0.00966574328348917,
+    "val_loss": 0.012790918740774355,
+    "lr": 0.0003461460113097137
+  },
+  {
+    "epoch": 181,
+    "train_data_loss": 0.012566334522016474,
+    "train_physics_loss": 0.009683210324969644,
+    "val_loss": 0.012804682644444608,
+    "lr": 0.00034117982500561395
+  },
+  {
+    "epoch": 182,
+    "train_data_loss": 0.012568229438288163,
+    "train_physics_loss": 0.009686671056037157,
+    "val_loss": 0.012741231111689441,
+    "lr": 0.00033623110995407757
+  },
+  {
+    "epoch": 183,
+    "train_data_loss": 0.012569256775448739,
+    "train_physics_loss": 0.009670571708588718,
+    "val_loss": 0.012761130876418563,
+    "lr": 0.00033130040883747687
+  },
+  {
+    "epoch": 184,
+    "train_data_loss": 0.012568165382027513,
+    "train_physics_loss": 0.00967244677549578,
+    "val_loss": 0.012804644655496362,
+    "lr": 0.0003263882623627533
+  },
+  {
+    "epoch": 185,
+    "train_data_loss": 0.012564852879695467,
+    "train_physics_loss": 0.009675207415302292,
+    "val_loss": 0.012829747223745911,
+    "lr": 0.00032149520920212257
+  },
+  {
+    "epoch": 186,
+    "train_data_loss": 0.012568226644320439,
+    "train_physics_loss": 0.00966888879986031,
+    "val_loss": 0.012733765810187305,
+    "lr": 0.00031662178593400343
+  },
+  {
+    "epoch": 187,
+    "train_data_loss": 0.012570396938716437,
+    "train_physics_loss": 0.009657042733688734,
+    "val_loss": 0.012828273405771582,
+    "lr": 0.00031176852698417556
+  },
+  {
+    "epoch": 188,
+    "train_data_loss": 0.012569057594355641,
+    "train_physics_loss": 0.009654363378726893,
+    "val_loss": 0.012750370014879492,
+    "lr": 0.00030693596456717455
+  },
+  {
+    "epoch": 189,
+    "train_data_loss": 0.012570215125817389,
+    "train_physics_loss": 0.009659837848335448,
+    "val_loss": 0.012824791017919779,
+    "lr": 0.00030212462862792697
+  },
+  {
+    "epoch": 190,
+    "train_data_loss": 0.012568476367777167,
+    "train_physics_loss": 0.00964902449342682,
+    "val_loss": 0.012766139048542227,
+    "lr": 0.00029733504678363775
+  },
+  {
+    "epoch": 191,
+    "train_data_loss": 0.012565512941757914,
+    "train_physics_loss": 0.009668478554917015,
+    "val_loss": 0.012805580769875837,
+    "lr": 0.0002925677442659297
+  },
+  {
+    "epoch": 192,
+    "train_data_loss": 0.012568144095916675,
+    "train_physics_loss": 0.009656304919445424,
+    "val_loss": 0.012816476958593535,
+    "lr": 0.0002878232438632462
+  },
+  {
+    "epoch": 193,
+    "train_data_loss": 0.012564962853516755,
+    "train_physics_loss": 0.009653905427800863,
+    "val_loss": 0.012767104289284157,
+    "lr": 0.00028310206586352245
+  },
+  {
+    "epoch": 194,
+    "train_data_loss": 0.012564814569997607,
+    "train_physics_loss": 0.009663503115159058,
+    "val_loss": 0.012777754000478213,
+    "lr": 0.00027840472799712883
+  },
+  {
+    "epoch": 195,
+    "train_data_loss": 0.012561272828003827,
+    "train_physics_loss": 0.009680622057596918,
+    "val_loss": 0.012734267973311005,
+    "lr": 0.0002737317453800964
+  },
+  {
+    "epoch": 196,
+    "train_data_loss": 0.01256364775005113,
+    "train_physics_loss": 0.009663965254385851,
+    "val_loss": 0.01273540600443319,
+    "lr": 0.0002690836304576292
+  },
+  {
+    "epoch": 197,
+    "train_data_loss": 0.01256600408982733,
+    "train_physics_loss": 0.00965447535657057,
+    "val_loss": 0.012762873726446302,
+    "lr": 0.00026446089294790756
+  },
+  {
+    "epoch": 198,
+    "train_data_loss": 0.012568393909501397,
+    "train_physics_loss": 0.009636861524868057,
+    "val_loss": 0.012801923128145357,
+    "lr": 0.0002598640397861931
+  },
+  {
+    "epoch": 199,
+    "train_data_loss": 0.012561793879126914,
+    "train_physics_loss": 0.009658920340084486,
+    "val_loss": 0.012767310411999784,
+    "lr": 0.00025529357506923715
+  },
+  {
+    "epoch": 200,
+    "train_data_loss": 0.012563493781399681,
+    "train_physics_loss": 0.009648639521432556,
+    "val_loss": 0.012756611070325297,
+    "lr": 0.0002507499999999999
+  },
+  {
+    "epoch": 201,
+    "train_data_loss": 0.0125607626752988,
+    "train_physics_loss": 0.009661456170638552,
+    "val_loss": 0.01277384395530868,
+    "lr": 0.0002462338128326895
+  },
+  {
+    "epoch": 202,
+    "train_data_loss": 0.012564208962913935,
+    "train_physics_loss": 0.009633585082568525,
+    "val_loss": 0.012763218601204215,
+    "lr": 0.00024174550881812148
+  },
+  {
+    "epoch": 203,
+    "train_data_loss": 0.012562560942554157,
+    "train_physics_loss": 0.009659916989547703,
+    "val_loss": 0.012717368471766672,
+    "lr": 0.0002372855801494095
+  },
+  {
+    "epoch": 204,
+    "train_data_loss": 0.01255587038833231,
+    "train_physics_loss": 0.0096673192385676,
+    "val_loss": 0.012735156778005822,
+    "lr": 0.00023285451590799124
+  },
+  {
+    "epoch": 205,
+    "train_data_loss": 0.012561445043155432,
+    "train_physics_loss": 0.009646862738216171,
+    "val_loss": 0.012794071402881415,
+    "lr": 0.0002284528020099941
+  },
+  {
+    "epoch": 206,
+    "train_data_loss": 0.012566751567205056,
+    "train_physics_loss": 0.00965749964725926,
+    "val_loss": 0.01276653767713616,
+    "lr": 0.0002240809211529496
+  },
+  {
+    "epoch": 207,
+    "train_data_loss": 0.012561750937557199,
+    "train_physics_loss": 0.009652604389603496,
+    "val_loss": 0.012777450043828256,
+    "lr": 0.00021973935276286082
+  },
+  {
+    "epoch": 208,
+    "train_data_loss": 0.012564063754349777,
+    "train_physics_loss": 0.009640854843441517,
+    "val_loss": 0.012826258774786707,
+    "lr": 0.00021542857294162636
+  },
+  {
+    "epoch": 209,
+    "train_data_loss": 0.012561366009743322,
+    "train_physics_loss": 0.009642925965237437,
+    "val_loss": 0.012742720341550247,
+    "lr": 0.00021114905441483182
+  },
+  {
+    "epoch": 210,
+    "train_data_loss": 0.012568731452965758,
+    "train_physics_loss": 0.009620835234108867,
+    "val_loss": 0.012797418275787946,
+    "lr": 0.00020690126647990968
+  },
+  {
+    "epoch": 211,
+    "train_data_loss": 0.012566558256801889,
+    "train_physics_loss": 0.009616752897355656,
+    "val_loss": 0.012788232355829208,
+    "lr": 0.00020268567495467478
+  },
+  {
+    "epoch": 212,
+    "train_data_loss": 0.012568403473671745,
+    "train_physics_loss": 0.009613061512854,
+    "val_loss": 0.012774757913223678,
+    "lr": 0.0001985027421262437
+  },
+  {
+    "epoch": 213,
+    "train_data_loss": 0.012561971402993918,
+    "train_physics_loss": 0.009629039867892437,
+    "val_loss": 0.012776878917770039,
+    "lr": 0.0001943529267003383
+  },
+  {
+    "epoch": 214,
+    "train_data_loss": 0.012564751860355177,
+    "train_physics_loss": 0.009604464088914742,
+    "val_loss": 0.012773211129129894,
+    "lr": 0.00019023668375098396
+  },
+  {
+    "epoch": 215,
+    "train_data_loss": 0.012560260423814322,
+    "train_physics_loss": 0.009640081824498,
+    "val_loss": 0.012802039986596472,
+    "lr": 0.00018615446467060623
+  },
+  {
+    "epoch": 216,
+    "train_data_loss": 0.012560262560377877,
+    "train_physics_loss": 0.009620018423564972,
+    "val_loss": 0.012758273999356935,
+    "lr": 0.0001821067171205294
+  },
+  {
+    "epoch": 217,
+    "train_data_loss": 0.012561543302105092,
+    "train_physics_loss": 0.00964567109949505,
+    "val_loss": 0.012832292708598318,
+    "lr": 0.0001780938849818867
+  },
+  {
+    "epoch": 218,
+    "train_data_loss": 0.01256431817329812,
+    "train_physics_loss": 0.00961217822673142,
+    "val_loss": 0.01280482461105191,
+    "lr": 0.00017411640830694239
+  },
+  {
+    "epoch": 219,
+    "train_data_loss": 0.012557548794195593,
+    "train_physics_loss": 0.009620827741115098,
+    "val_loss": 0.012780640739947557,
+    "lr": 0.0001701747232708357
+  },
+  {
+    "epoch": 220,
+    "train_data_loss": 0.012555646699494497,
+    "train_physics_loss": 0.009623299969582902,
+    "val_loss": 0.012761775006149565,
+    "lr": 0.00016626926212375023
+  },
+  {
+    "epoch": 221,
+    "train_data_loss": 0.012564418939926592,
+    "train_physics_loss": 0.009645834695936606,
+    "val_loss": 0.012806181270148485,
+    "lr": 0.00016240045314351093
+  },
+  {
+    "epoch": 222,
+    "train_data_loss": 0.012564384834434434,
+    "train_physics_loss": 0.009610871722536477,
+    "val_loss": 0.012818366131414809,
+    "lr": 0.0001585687205886199
+  },
+  {
+    "epoch": 223,
+    "train_data_loss": 0.012561673795065686,
+    "train_physics_loss": 0.009622138456584606,
+    "val_loss": 0.012825323817049783,
+    "lr": 0.0001547744846517318
+  },
+  {
+    "epoch": 224,
+    "train_data_loss": 0.012560309801582932,
+    "train_physics_loss": 0.0096041439560576,
+    "val_loss": 0.012781378948279926,
+    "lr": 0.00015101816141357386
+  },
+  {
+    "epoch": 225,
+    "train_data_loss": 0.012557978939752764,
+    "train_physics_loss": 0.009620689763996814,
+    "val_loss": 0.012780930388778929,
+    "lr": 0.0001473001627973195
+  },
+  {
+    "epoch": 226,
+    "train_data_loss": 0.012554776764685107,
+    "train_physics_loss": 0.009634126350637066,
+    "val_loss": 0.012747884134671861,
+    "lr": 0.0001436208965234148
+  },
+  {
+    "epoch": 227,
+    "train_data_loss": 0.012562744216940435,
+    "train_physics_loss": 0.009616599745157774,
+    "val_loss": 0.012755914260783503,
+    "lr": 0.00013998076606486806
+  },
+  {
+    "epoch": 228,
+    "train_data_loss": 0.012551398917792304,
+    "train_physics_loss": 0.00961962075766169,
+    "val_loss": 0.01274930329753026,
+    "lr": 0.000136380170603005
+  },
+  {
+    "epoch": 229,
+    "train_data_loss": 0.012545378489925242,
+    "train_physics_loss": 0.00965345524201979,
+    "val_loss": 0.01274903422792352,
+    "lr": 0.00013281950498369283
+  },
+  {
+    "epoch": 230,
+    "train_data_loss": 0.01254561161392667,
+    "train_physics_loss": 0.009694458229336499,
+    "val_loss": 0.012743048384905822,
+    "lr": 0.0001292991596740415
+  },
+  {
+    "epoch": 231,
+    "train_data_loss": 0.012563762106566886,
+    "train_physics_loss": 0.009621221678193652,
+    "val_loss": 0.012761433623851307,
+    "lr": 0.00012581952071958542
+  },
+  {
+    "epoch": 232,
+    "train_data_loss": 0.012558934234605794,
+    "train_physics_loss": 0.009613977851208292,
+    "val_loss": 0.01275955987793784,
+    "lr": 0.00012238096970194773
+  },
+  {
+    "epoch": 233,
+    "train_data_loss": 0.012556878701417903,
+    "train_physics_loss": 0.009596675535420087,
+    "val_loss": 0.012774631268374862,
+    "lr": 0.00011898388369699619
+  },
+  {
+    "epoch": 234,
+    "train_data_loss": 0.012546250100054935,
+    "train_physics_loss": 0.009639023518652798,
+    "val_loss": 0.012710550619709876,
+    "lr": 0.00011562863523349331
+  },
+  {
+    "epoch": 235,
+    "train_data_loss": 0.012549169787490277,
+    "train_physics_loss": 0.009680801592555173,
+    "val_loss": 0.012776491487578999,
+    "lr": 0.00011231559225224303
+  },
+  {
+    "epoch": 236,
+    "train_data_loss": 0.012562674321799736,
+    "train_physics_loss": 0.009603306532711978,
+    "val_loss": 0.01278478076921836,
+    "lr": 0.00010904511806574305
+  },
+  {
+    "epoch": 237,
+    "train_data_loss": 0.012564395768196804,
+    "train_physics_loss": 0.009585503119431026,
+    "val_loss": 0.012777400623646474,
+    "lr": 0.00010581757131834274
+  },
+  {
+    "epoch": 238,
+    "train_data_loss": 0.012566335792644427,
+    "train_physics_loss": 0.009573426915852236,
+    "val_loss": 0.012814718831871305,
+    "lr": 0.00010263330594691395
+  },
+  {
+    "epoch": 239,
+    "train_data_loss": 0.012566983690962506,
+    "train_physics_loss": 0.009562791848247711,
+    "val_loss": 0.012780786792357121,
+    "lr": 9.949267114203836e-05
+  },
+  {
+    "epoch": 240,
+    "train_data_loss": 0.012563339902876225,
+    "train_physics_loss": 0.009571183511750521,
+    "val_loss": 0.012768019163500397,
+    "lr": 9.639601130971379e-05
+  },
+  {
+    "epoch": 241,
+    "train_data_loss": 0.012561630485915142,
+    "train_physics_loss": 0.009587230259796473,
+    "val_loss": 0.012775578791455876,
+    "lr": 9.33436660335871e-05
+  },
+  {
+    "epoch": 242,
+    "train_data_loss": 0.012564593415346837,
+    "train_physics_loss": 0.009576790550797895,
+    "val_loss": 0.012806031499959288,
+    "lr": 9.033597003771482e-05
+  },
+  {
+    "epoch": 243,
+    "train_data_loss": 0.012560488124231347,
+    "train_physics_loss": 0.009577725001343965,
+    "val_loss": 0.01276443163383632,
+    "lr": 8.737325314985631e-05
+  },
+  {
+    "epoch": 244,
+    "train_data_loss": 0.012564108044304929,
+    "train_physics_loss": 0.00956923715183529,
+    "val_loss": 0.012757675369240103,
+    "lr": 8.445584026530532e-05
+  },
+  {
+    "epoch": 245,
+    "train_data_loss": 0.012561287958903036,
+    "train_physics_loss": 0.009575132655237838,
+    "val_loss": 0.01273438233071037,
+    "lr": 8.158405131126074e-05
+  },
+  {
+    "epoch": 246,
+    "train_data_loss": 0.01256196231067011,
+    "train_physics_loss": 0.009575422133974717,
+    "val_loss": 0.01276774806601386,
+    "lr": 7.875820121174347e-05
+  },
+  {
+    "epoch": 247,
+    "train_data_loss": 0.012562686178048602,
+    "train_physics_loss": 0.009565357829265621,
+    "val_loss": 0.012774472860919853,
+    "lr": 7.597859985306155e-05
+  },
+  {
+    "epoch": 248,
+    "train_data_loss": 0.012561532746526842,
+    "train_physics_loss": 0.00957453144852079,
+    "val_loss": 0.012779008852498185,
+    "lr": 7.324555204982696e-05
+  },
+  {
+    "epoch": 249,
+    "train_data_loss": 0.012560488456467863,
+    "train_physics_loss": 0.00956939766448578,
+    "val_loss": 0.012741045808539755,
+    "lr": 7.055935751153021e-05
+  },
+  {
+    "epoch": 250,
+    "train_data_loss": 0.012563669025558674,
+    "train_physics_loss": 0.0095669493411422,
+    "val_loss": 0.012785721825616013,
+    "lr": 6.792031080967299e-05
+  },
+  {
+    "epoch": 251,
+    "train_data_loss": 0.012566145953052523,
+    "train_physics_loss": 0.00954334857737543,
+    "val_loss": 0.012762359253341151,
+    "lr": 6.532870134546531e-05
+  },
+  {
+    "epoch": 252,
+    "train_data_loss": 0.01256348476153171,
+    "train_physics_loss": 0.009550327709054245,
+    "val_loss": 0.012758324328329294,
+    "lr": 6.278481331809015e-05
+  },
+  {
+    "epoch": 253,
+    "train_data_loss": 0.012557188676320285,
+    "train_physics_loss": 0.009566810968167642,
+    "val_loss": 0.012771309653837834,
+    "lr": 6.028892569353643e-05
+  },
+  {
+    "epoch": 254,
+    "train_data_loss": 0.012547107839310078,
+    "train_physics_loss": 0.009587820066206256,
+    "val_loss": 0.01274998505569754,
+    "lr": 5.784131217400825e-05
+  },
+  {
+    "epoch": 255,
+    "train_data_loss": 0.012537077875132805,
+    "train_physics_loss": 0.009621674184328691,
+    "val_loss": 0.012726507082040752,
+    "lr": 5.5442241167910304e-05
+  },
+  {
+    "epoch": 256,
+    "train_data_loss": 0.012517884861416338,
+    "train_physics_loss": 0.009659578440537716,
+    "val_loss": 0.012707992471874721,
+    "lr": 5.309197576041338e-05
+  },
+  {
+    "epoch": 257,
+    "train_data_loss": 0.012500853555195698,
+    "train_physics_loss": 0.009763489088548995,
+    "val_loss": 0.012685068920555135,
+    "lr": 5.0790773684604345e-05
+  },
+  {
+    "epoch": 258,
+    "train_data_loss": 0.01248387610663499,
+    "train_physics_loss": 0.009742813785710642,
+    "val_loss": 0.012686459602968346,
+    "lr": 4.853888729322334e-05
+  },
+  {
+    "epoch": 259,
+    "train_data_loss": 0.01245969261062213,
+    "train_physics_loss": 0.010028749624836264,
+    "val_loss": 0.012707418882318081,
+    "lr": 4.633656353098928e-05
+  },
+  {
+    "epoch": 260,
+    "train_data_loss": 0.012446596854467093,
+    "train_physics_loss": 0.009733132233373128,
+    "val_loss": 0.01263385670890491,
+    "lr": 4.418404390752093e-05
+  },
+  {
+    "epoch": 261,
+    "train_data_loss": 0.012411894137128142,
+    "train_physics_loss": 0.009971594485671628,
+    "val_loss": 0.01261922303256729,
+    "lr": 4.208156447085154e-05
+  },
+  {
+    "epoch": 262,
+    "train_data_loss": 0.012373231908048466,
+    "train_physics_loss": 0.009706654013320097,
+    "val_loss": 0.012565121084692018,
+    "lr": 4.002935578154395e-05
+  },
+  {
+    "epoch": 263,
+    "train_data_loss": 0.01234403953656017,
+    "train_physics_loss": 0.00990815923869101,
+    "val_loss": 0.012533020250679504,
+    "lr": 3.802764288740764e-05
+  },
+  {
+    "epoch": 264,
+    "train_data_loss": 0.012310336872341511,
+    "train_physics_loss": 0.009720275894273956,
+    "val_loss": 0.0125011803822652,
+    "lr": 3.607664529881846e-05
+  },
+  {
+    "epoch": 265,
+    "train_data_loss": 0.012273652401295061,
+    "train_physics_loss": 0.009668816008253388,
+    "val_loss": 0.01245647818300753,
+    "lr": 3.4176576964647736e-05
+  },
+  {
+    "epoch": 266,
+    "train_data_loss": 0.012255545161254492,
+    "train_physics_loss": 0.009577174531384257,
+    "val_loss": 0.012462544202383968,
+    "lr": 3.2327646248800225e-05
+  },
+  {
+    "epoch": 267,
+    "train_data_loss": 0.012237864095837839,
+    "train_physics_loss": 0.009612891158580328,
+    "val_loss": 0.012424174628611053,
+    "lr": 3.053005590736439e-05
+  },
+  {
+    "epoch": 268,
+    "train_data_loss": 0.01221877178336287,
+    "train_physics_loss": 0.009484990399182408,
+    "val_loss": 0.012435488823440767,
+    "lr": 2.8784003066378248e-05
+  },
+  {
+    "epoch": 269,
+    "train_data_loss": 0.012213117591471215,
+    "train_physics_loss": 0.009444630079225864,
+    "val_loss": 0.012405834213319806,
+    "lr": 2.708967920021202e-05
+  },
+  {
+    "epoch": 270,
+    "train_data_loss": 0.012200282220384088,
+    "train_physics_loss": 0.009369950859932905,
+    "val_loss": 0.012405735034976275,
+    "lr": 2.5447270110570818e-05
+  },
+  {
+    "epoch": 271,
+    "train_data_loss": 0.012189794144464172,
+    "train_physics_loss": 0.009334882708934384,
+    "val_loss": 0.012385271279321562,
+    "lr": 2.3856955906119732e-05
+  },
+  {
+    "epoch": 272,
+    "train_data_loss": 0.012179299010285503,
+    "train_physics_loss": 0.009304510786479066,
+    "val_loss": 0.012379822869514745,
+    "lr": 2.2318910982732418e-05
+  },
+  {
+    "epoch": 273,
+    "train_data_loss": 0.012174734494081598,
+    "train_physics_loss": 0.00931169850967358,
+    "val_loss": 0.012387913967212361,
+    "lr": 2.0833304004366946e-05
+  },
+  {
+    "epoch": 274,
+    "train_data_loss": 0.01217000006894233,
+    "train_physics_loss": 0.009215258682178479,
+    "val_loss": 0.012363576578096516,
+    "lr": 1.9400297884569785e-05
+  },
+  {
+    "epoch": 275,
+    "train_data_loss": 0.012163367641973089,
+    "train_physics_loss": 0.00929188886851751,
+    "val_loss": 0.012365393723631579,
+    "lr": 1.8020049768610386e-05
+  },
+  {
+    "epoch": 276,
+    "train_data_loss": 0.012160211995922637,
+    "train_physics_loss": 0.009232401593360322,
+    "val_loss": 0.012376544080794818,
+    "lr": 1.6692711016248786e-05
+  },
+  {
+    "epoch": 277,
+    "train_data_loss": 0.012155191456824824,
+    "train_physics_loss": 0.009189125684141208,
+    "val_loss": 0.012355275654924973,
+    "lr": 1.541842718513684e-05
+  },
+  {
+    "epoch": 278,
+    "train_data_loss": 0.01214943144230112,
+    "train_physics_loss": 0.009199477999031092,
+    "val_loss": 0.012333209656419293,
+    "lr": 1.4197338014856439e-05
+  },
+  {
+    "epoch": 279,
+    "train_data_loss": 0.012148823672145662,
+    "train_physics_loss": 0.009178403966425719,
+    "val_loss": 0.012365536891945427,
+    "lr": 1.3029577411595715e-05
+  },
+  {
+    "epoch": 280,
+    "train_data_loss": 0.012147674480173913,
+    "train_physics_loss": 0.00916736428776095,
+    "val_loss": 0.012368466449721206,
+    "lr": 1.1915273433464115e-05
+  },
+  {
+    "epoch": 281,
+    "train_data_loss": 0.012150342272244662,
+    "train_physics_loss": 0.00913624005812791,
+    "val_loss": 0.012361533376538465,
+    "lr": 1.0854548276449986e-05
+  },
+  {
+    "epoch": 282,
+    "train_data_loss": 0.012143102767868445,
+    "train_physics_loss": 0.009125259238342972,
+    "val_loss": 0.01235531974253395,
+    "lr": 9.847518261020041e-06
+  },
+  {
+    "epoch": 283,
+    "train_data_loss": 0.012143093726793888,
+    "train_physics_loss": 0.00913287860922784,
+    "val_loss": 0.012347185255719288,
+    "lr": 8.894293819363682e-06
+  },
+  {
+    "epoch": 284,
+    "train_data_loss": 0.01214243771695661,
+    "train_physics_loss": 0.009113239430145588,
+    "val_loss": 0.012351732445700515,
+    "lr": 7.994979483282762e-06
+  },
+  {
+    "epoch": 285,
+    "train_data_loss": 0.012138940259616334,
+    "train_physics_loss": 0.009115727692559729,
+    "val_loss": 0.012347753685448439,
+    "lr": 7.14967387272874e-06
+  },
+  {
+    "epoch": 286,
+    "train_data_loss": 0.012139337658670075,
+    "train_physics_loss": 0.00910313187990571,
+    "val_loss": 0.012353702545947125,
+    "lr": 6.358469684987329e-06
+  },
+  {
+    "epoch": 287,
+    "train_data_loss": 0.012141322937972519,
+    "train_physics_loss": 0.009085535002868813,
+    "val_loss": 0.012357413828853638,
+    "lr": 5.621453684513923e-06
+  },
+  {
+    "epoch": 288,
+    "train_data_loss": 0.012142050567752836,
+    "train_physics_loss": 0.00908913350252318,
+    "val_loss": 0.012352905423951245,
+    "lr": 4.938706693418357e-06
+  },
+  {
+    "epoch": 289,
+    "train_data_loss": 0.01213954741654801,
+    "train_physics_loss": 0.009067898225640096,
+    "val_loss": 0.012343698887214545,
+    "lr": 4.310303582601976e-06
+  },
+  {
+    "epoch": 290,
+    "train_data_loss": 0.012140617850793267,
+    "train_physics_loss": 0.0090759754113754,
+    "val_loss": 0.012356166705308903,
+    "lr": 3.736313263547492e-06
+  },
+  {
+    "epoch": 291,
+    "train_data_loss": 0.012140234082272768,
+    "train_physics_loss": 0.009054208329862843,
+    "val_loss": 0.012343051678110515,
+    "lr": 3.2167986807615425e-06
+  },
+  {
+    "epoch": 292,
+    "train_data_loss": 0.012134726148671286,
+    "train_physics_loss": 0.009080221692711624,
+    "val_loss": 0.012347633867794949,
+    "lr": 2.7518168048725676e-06
+  },
+  {
+    "epoch": 293,
+    "train_data_loss": 0.012136090743007312,
+    "train_physics_loss": 0.009045956749999206,
+    "val_loss": 0.01234223744681766,
+    "lr": 2.3414186263831814e-06
+  },
+  {
+    "epoch": 294,
+    "train_data_loss": 0.012138092240046505,
+    "train_physics_loss": 0.009071834768880816,
+    "val_loss": 0.012351111508905888,
+    "lr": 1.9856491500783564e-06
+  },
+  {
+    "epoch": 295,
+    "train_data_loss": 0.012138003587680366,
+    "train_physics_loss": 0.009048133961276718,
+    "val_loss": 0.012341080714137323,
+    "lr": 1.6845473900903703e-06
+  },
+  {
+    "epoch": 296,
+    "train_data_loss": 0.012138034916876955,
+    "train_physics_loss": 0.009059383431396516,
+    "val_loss": 0.012349864498021142,
+    "lr": 1.4381463656202375e-06
+  },
+  {
+    "epoch": 297,
+    "train_data_loss": 0.012137890109470719,
+    "train_physics_loss": 0.009027834753709116,
+    "val_loss": 0.012339886683490007,
+    "lr": 1.2464730973170658e-06
+  },
+  {
+    "epoch": 298,
+    "train_data_loss": 0.012135721260657007,
+    "train_physics_loss": 0.009058773598900557,
+    "val_loss": 0.012341006148246026,
+    "lr": 1.109548604314673e-06
+  },
+  {
+    "epoch": 299,
+    "train_data_loss": 0.012135434980580204,
+    "train_physics_loss": 0.009045215629295673,
+    "val_loss": 0.012339619536613744,
+    "lr": 1.0273879019266845e-06
+  },
+  {
+    "epoch": 300,
+    "train_data_loss": 0.012133789656991299,
+    "train_physics_loss": 0.009055665649996762,
+    "val_loss": 0.012341830947045837,
+    "lr": 1e-06
+  }
+]